Ubuntu服务器上安装pytorch环境
任务描述
Linux系统版本:Ubuntu16.04
cuda==9.2
conda == 4.8.0
python==3.6.4
pytorch==1.6.0
此过程是自己在window和linux安装时环境记录的过程后续整合的一个小tips,方便之后复盘
安装步骤
-
conda的因为服务器上已经装好了,这里主要记录一下环境的创建,包的安装
查看conda版本
conda --version
更新conda
conda update conda
-
首先到CUDA官网找到9.2版本,我这里是直接下载run文件到本地,然后在上传到服务器上,然后在服务器中使用sh命令直接运行这个run文件
下载链接:
-
安装好后,如果遇到CUDA多版本的问题,参考上一篇笔记
-
设置清华的镜像源,原本的anaconda用的是外网的源,会导致下载速度特别慢和频率很高的掉包
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
-
然后按照如下修改.condarc文件
-
使用conda创建虚拟环境,我这里使用的pytorch版本是1.6.0,命令为py364-pt16,其中py364表示python版本是3.6.4,pt16表示pytorch版本是1.6.0
conda create -n py364-pt16 python=3.6.4
过程中会提示导入库,直接y后回车即可
查看环境是否创建成功
conda info -e
查看安装环境中的包,-n限定查看虚拟环境
conda list -n py364-pt16
激活环境
activate py364-pt16
退出当前环境
conda deactivate
-
安装pythorch
这里记录一个当时遇到的问题,当时是直接使用官网的conda命令安装的,安装好后查看的pytorch版本不是1.6.0
conda install pytorch torchvison cudatoolkit=9.2 pytorch
后续使用pip又重新安装了对应的版本
pip install torch==1.6.0 torchvision==0.7.0
查看torch版本
pip list |grep pytorch
版本是相应版本,但是试着运行一段代码遇到了pytorch版本1.6和cuda版本9.2不匹配的问题
解决办法:先将pytorch、torchvision都卸载掉,使用官网上的pip重新安装后
pip install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
考虑下载速率的问题,我因为之前加了清华的镜像源就想着不直接用-f后面链接下载,但是命令提示报错,网友测试这里不用用国内的镜像源
至此,环境的需要安装的关键部分就已经成功了,根据这个问题总结一下:之后安装一些库首先还是官网相应的版本安装命令提示,找的一些博客上的命令会根据使用情况不同缺少一些东西导致后期的一些错误
-
另外可能会根据程序的使用去安装一些工具包,一般地,版本都没有大要求,版本要是对应不上也会有相关的提示,通常按照以下命令可以安装大部分需要的工具包
pip install packagename(包名)
版权声明:本文为Vicky1361原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。