服务器上安装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
可以运行了。