cuda是高级版本的,直接去官网查一下命令,轻易安装,但是cuda是低版本的,比如8.0时,安装就不那么简单。
1.首先,创建一个python=3.6的虚拟环境
conda create -n env_name python=3.6
(系统应该会默认安装python3.6.13的)
2.在虚拟环境中安装pytorch
conda install pytorch1.0.0 torchvision0.2.1 cuda80 -c pytorch
(这条命令是怎么来的呢?
法一:上网查,总有好心人贴出来。
法二:官网查看 install previous versions of PyTorch找到对应的cuda8.0版本的pytorch安装命令)
可见,上述安装过程也不难,但是安装过程中,我遇到过一些问题:
问1:404 (具体问题不记得了,大概意思就是下载源好像下载不了)
明明是我添加的那些常用的下载源,结果竟然下载不了,所以我:
conda config --remove-key channels
删除了所有添加的下载源
问题2:清除了添加的下载源,使用默认下载源下载,结果下载到一般,出现错误:CondaError: Failed to write to /home/gpu_server/anaconda3/…
所以我就到anaconda3所在的目录下:
sudo chown -R username anaconda3
问3:然后我开始重新执行pytorch安装命令,结果显示
反正一大串,也看不懂,最后问我是否提交错误。
所以我执行了(应该是为了清除上一次安装的那些残留的包)
conda clean -i
然后在执行pytorch安装命令
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
安装成功!开心!