网上教程的镜像源很多莫名的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键就可以完成下载了.
使用镜像源