整理pytorch环境搭建的相关的内容
安装cuda和cudnn
参考链接:cuda安装
如果没有GPU,忽略此步骤
安装anaconda
在anaconda官网下载individual edition
安装pytorch
- 添加源:在anaconda prompt中输入如下指令
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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- 创建pytorch环境:相当于为pytorch新建一个房间。如果不进行这一步的话,就很有可能把pytorch安装在base环境下,虽然也不会报错,但很容易把环境弄乱
conda create -n env_name python=x.x(3.7)
使用 conda env list
或conda info --envs
查看pytorch环境是否创建成功,创建成功可以看到名为base环境下会列出刚创建的pytorch环境
- 激活pytorch环境:完成此步之后才可继续进行下一步安装pytorch,否则pytorch将被装在base环境下
conda activate pytorch
- 安装pytorch:在pytorch官网选择电脑的对应配置,复制对应的安装命令
- 测试pytorch环境:在命令行中
import pytorch
测试是否安装成功
pycharm中配置pytorch
-
添加解释器:
settings
—project interpreter
—Add
-
在
conda environment
下找到pytorch虚拟环境
-
anaconda下添加的pytorch虚拟环境位置查看 :
(base) C:\Users\km>conda env list
-
如果报错的话,可以点图中的
Show All
,把红的删除即可
jupyter notebook 内核启动时报错
juypter 启动时提示 KernelRestarter: restart failed,猜测应该是版本的问题,折腾一顿之后还是重装了ipykernel可以了