win10安装pytorch

网上教程的镜像源很多莫名的debug,官网pip命令又下载很慢.无数崩溃后发现的一个较好的方式安装pytorch。前置：anconda prompt

1.anconda prompt shell

2.添加清华源

方法一:在.anconda prompt shell中依次输入 假设是64位要在后面加win-64 ,不是64位就不用再加

# 添加清华镜像 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64

# 添加pytorch镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

# for legacy win-64 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/win-64

conda config --set show_channel_urls yes

这时候谨记打开c盘/用户名/.condarc文件,记事本的形式打开

默认在anconda prompt  shell修改源后还存在 -defult 我们需要把这一行delete掉

方法2:直接在.condarc文件中修改源文件

将.condarc中的命名行替换成

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

直接保存生效

3.安装pytorch

两种情况:在电脑中直接安装pytorch,在conda虚拟环境中安装pytorch

不建议在电脑环境中直接安装,假设你需要多环境框架,譬如tensorflow框架就会存在debug

步骤:

(1)同样打开我们的anconda prompt  shell

conda create --name pytorch_gpu python=3.8

此时name可以自定义,我们这里自定义的是:pytorch, python版本对应是3.8

大家可以一定要注意自己在接下来的cuda版本与Python版本相对应

(2)查看自己电脑的cuda版本:

打开NVIDIA控制面板设置

帮助->系统信息->组件

博主这里的是 cuda10.2 ,向下兼容,倘若是10.1则对应python版本是3.7 具体的对应环境变量对应可以自行去pytorch官网查询

(3)进入conda虚拟环境:conda activate 虚拟环境名

conda activate pytorch_gpu

(4)键入命令:下载pytorch

conda install pytorch torchvision cudatoolkit=10.2

官网默认的命令是:conda install pytorch torchvision cudatoolkit=10.2 -c -python,这里我们把-c -python去掉

出现命令提示按y键就可以完成下载了.

使用镜像源