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

【jupyter notebook中创建环境、安装使用pytorch】注意不要官方命令原版的结尾的-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