一、搭建pytorch环境
conda create -n pytorch python=3.7
二、安装pytorch
激活新建的pytorch环境
conda activate pytorch
安装最新的pytorch
pip3 install torch===1.3.0 torchvision===0.4.1 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
#从清华镜像源下载安装,经检验十分的快,此处默认是使用 cuda10.1
pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
#cuda9.2
pip3 install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
#cpu版本
验证输入python 进入
import torch
torch.__version__# 得到结果'1.3.0'
三、配置 Jupyter Notebook
新建的环境是没有安装安装ipykernel的所以无法注册到Jupyter Notebook中,所以先要准备下环境
conda install ipykernel#安装ipykernel(还是在pytorch 环境下)
python -m ipykernel install --user --name pytorch --display-name Pytorch-for-Deeplearning#写入环境
下一步就是定制 Jupyter Notebook
conda activate base#切换回基础环境
jupyter notebook --generate-config#创建jupyter notebook配置文件,这里会显示创建jupyter_notebook_config.py的具体位置。
打开文件,修改
c.NotebookApp.notebook_dir = '' 默认目录位置
c.NotebookApp.iopub_data_rate_limit = 100000000 这个改大一些否则有可能报错
至此 pytorch的开发环境安装完成,可以在开始菜单中打开Jupyter Notebook 在New 菜单中创建文件时选择Pytorch-for-Deeplearning 创建pytorch的相关开发环境了,输入如下语句,验证可得即安装成功。
配置pycharm环境即将下图的内核换为envs/pytorch环境下的python内核即可。
如有遇到安装问题即可给我留言。
网友评论