服务器上安装pytorch环境以及遇到的一些问题
·
服务器上安装pytorch环境以及遇到的一些问题
此文章只是给自己做个解决问题的记录。
首先使用的是云服务器,conda 已经装好的了,后面需要安装一个pytorch 的cuda 版本。
为了和代码中的版本契合,我们选择直接安装指定的版本,避免其他问题的出现
安装的指令:
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html
就可以安装好了
然后 conda list查看一下:

安装成功了。
这里还遇到了一个bug就是ptotobuf的版本问题,
报错如下:

我的做法:
pip uninstall protobuf
再次安装指定版本:
conda install protobuf=3.19.0
可以运行了。
更多推荐



所有评论(0)