任务描述
Linux系统版本:Ubuntu16.04
cuda==9.2
conda == 4.8.0
python==3.6.4
pytorch==1.6.0
此过程是自己在window和linux安装时环境记录的过程后续整合的一个小tips,方便之后复盘
安装步骤
-
使用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后面链接下载,但是命令提示报错,网友测试这里不用用国内的镜像源
至此,环境的需要安装的关键部分就已经成功了,根据这个问题总结一下:之后安装一些库首先还是官网相应的版本安装命令提示,找的一些博客上的命令会根据使用情况不同缺少一些东西导致后期的一些错误
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步