jupyter notebook中创建环境、安装使用pytorch

5.8 torch环境

Ref：https://zhuanlan.zhihu.com/p/101799677

• 添加镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

• anaconda prompt创建新环境

conda create -n torch

• 进入环境

conda activate torch

• 进入pytorch官网https://pytorch.org/

conda install pytorch torchvision cpuonly

注意不要官方命令原版的结尾的-c pytorch

• 验证pytorch安装好

python
import torch
torch.__version__

• 新建的环境是没有安装ipykernel的，所以无法注册到Jupyter Notebook中，所以先要准备下环境：

# 安装ipykernel
conda install ipykernel
# 写入环境
python -m ipykernel install  --name torch --display-name "Pytorch for Deeplearning"

# 切换回基础环境
activate base
# 创建jupyter notebook配置文件
jupyter notebook --generate-config
## 这里会显示创建jupyter_notebook_config.py的具体位置

• 打开文件，修改

c.NotebookApp.notebook_dir = '' 默认目录位置
c.NotebookApp.iopub_data_rate_limit = 100000000 这个改大一些否则有可能报错

• 不明原因的，kernel中多出了一个Pytorch for Deeplearning（不可用）和一个Python[conda env:torch]可用

5.9 pytorch环境

Ref：https://zhuanlan.zhihu.com/p/101799677
待参考：https://blog.csdn.net/u014665013/article/details/81084604

• anaconda新建环境，等价于

conda create -n pytorch

• 昨天已经在base环境中

conda install ipykernel

• 进入新环境，创建kernel文件，将环境写入notebook的kernel中

conda activate pytorch
conda install -n pytorch ipykernel
python -m ipykernel install --user --name pytorch --display-name "pytorch"

• 安装pytorch

conda install pytorch torchvision cpuonly

至此，完毕

总结

• pytorch和torch两个虚拟环境都装了pytorch
• 在jupyter notebook中新建需要运行pytorch时，选择env:pytorch或env:torch中的任意一个，不要选pytorch for deep learning. 没有pytorch可选
• 在ipynb文件中，需要用到pytorch时，选择kernel为env:pytorch或env:torch中的任意一个，不要选pytorch for deep learning，也不要选pytorch