pytorch 环境搭建

pytorch 环境搭建

网上博客很多，但是有些错漏或者表达过于复杂，导致遇到些坑。
这篇是针对 python 3.7 + pytorch 1.2 + cuda 10.0 当前最新版本最简洁的说明。

安装cuda + cudnn

安装anaconda

修改C:\Users\Administrator.condarc为以下内容

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
ssl_verify: false
show_channel_urls: true

初始化环境

打开anaconda prompt命令行
conda create -n pytorch
conda activate pytorch

安装pytorch

conda install pytorch torchvision

测试pytorch（gpu版本）如下图

jupyter conda 环境创建

安装ipykernel

conda install ipykernel

注册ipykernel到anaconda的base环境

python -m ipykernel install --user --name pytorch --display-name “Python [conda env:pytorch]”

在base的环境下jupyter notebook启动，网页上新建时即可选择pytorch环境

vscode conda 环境创建

打开文件夹，并保存到workspace
Ctrl+Shift+P调出命令框
打开terminal profile的配置，选择Command Prompt 【即cmd.exe】
打开python的配置，选择对应conda envs 下的python.exe的全路径即可
参考链接

nvidia下载

cuda不同版本归档路径
cudnn不同版本归档路径