One Lab实验室服务器配置

One Lab实验室服务器配置

文章目录

• One Lab实验室服务器配置
• • • Pytorch和CUDA配置
    • Conda重新安装

Pytorch和CUDA配置

nvcc -V查看CUDA版本为9.0

踩坑记录： 使用pytorch官网的whl包下载 出现问题，安装一半断开出错
安装1.7版本pytorch和9.0版本CUDA的命令为：

pip install torch==1.7.1+cu92 torchvision==0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

成功记录： 更换conda源为清华源，重新安装即成功

换源以后运行：

conda install pytorch==1.7.1 torchvision==0.8.2 cuda9.2

安装成功
参考推文

方法二：使用豆瓣源更加稳定

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple some-package

参考链接

安装失败需要重试，可能要换个时间重试。

Conda重新安装

重新安装Anaconda，安装教程