anaconda配置pytorch环境
时间: 2023-09-22 13:08:58 浏览: 145
STD-anaconda配置pytorch环境
如果要在Anaconda中配置PyTorch环境,可以使用以下步骤:
1. 打开Anaconda prompt。
2. 输入conda create --name 环境名称,例如conda create --name pytorch_env
3. 输入conda activate pytorch_env,激活新建的环境。
4. 输入conda install pytorch torchvision -c pytorch,安装PyTorch和torchvision。
5. 输入conda install -c anaconda ipykernel,安装ipykernel
6. 输入python -m ipykernel install --user --name=pytorch_env,将新建的环境添加到jupyter notebook中
7. 打开jupyter notebook 检查是否成功
要注意, pytorch安装依赖于你使用的版本,cuda,pytorch版本,平台,等等。需要注意上面的命令可能不能在所有环境中正常工作,根据需要进行调整。
阅读全文