PyTorch安装

ZH2333

于 2020-12-28 14:14:43 发布

阅读量605

点赞数 1

文章标签： python pytorch

本文链接：https://blog.csdn.net/ZH2333/article/details/111837033

版权

参考中文教程：https://github.com/zergtant/pytorch-handbook

创建新的Anacoda环境。

conda create --name mytorch python=3.8

激活：

conda activate mytorch $AppData\Local\Temp$

安装pytorch

CPU版本+清华镜像

pip install torch==1.5.1+cpu torchvision==0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述
进入Python测试;

在这里插入图片描述

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

使用conda ipykernel没有成功

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

在这里插入图片描述
订制Jupyter Notebook

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

在这里插入图片描述
打开文件，修改

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

在这里插入图片描述
“#”要去掉

打开jupyter notebook 点击右侧的Pytorch for Deeplearning

配置完成。

关注