1.确定自己的CUDA版本
cat /usr/local/cuda/version.txt
2.建立虚拟环境
conda create -n pytorch
开启、退出、删除命令:
source activate pytorch
source deactivate
conda remove -n pytorch --all
3.安装pytorch
https://pytorch.org/ 在官网主页根据你的系统和CUDA,python版本,选择conda安装方式。
使用官方指定的方法下载太慢,所以:
- 进入anaconda 的清华镜像帮助:网站为:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
- 复制以下命令,粘贴回车
conda config --add channels https://mirrors.