之前安装过cuda10,pytorch主推的1.7不支持cuda10,只能安装1.2版本
1.创建conda虚拟环境
conda create -n pytorch python=2.7.17
因为之前试过3版本的python,在安装时报错,所以现在采用2版本的
2.更换源
设置清华源镜像
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
设置pytorch镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
3.按照官网命令安装
conda install pytorch1.2.0 torchvision0.4.0 cudatoolkit=10.0
去掉最后的 -c pytorch 快的飞起