One Lab实验室服务器配置

Enabler_Rx

已于 2022-04-24 10:41:50 修改

阅读量1.9k

点赞数 1

分类专栏：编程机器学习文章标签：深度学习

于 2022-04-23 00:42:21 首次发布

本文链接：https://blog.csdn.net/weixin_42305039/article/details/124354243

版权

编程同时被 2 个专栏收录

31 篇文章 0 订阅

订阅专栏

机器学习

21 篇文章 0 订阅

订阅专栏

本文记录了在OneLab实验室中配置Pytorch、CUDA的过程，遇到的问题及解决方案。首先，通过nvcc-V检查到CUDA版本为9.0。尝试直接使用pytorch官网whl包安装失败，后改用清华源通过conda成功安装了Pytorch 1.7.1、CUDA 9.2。此外，还介绍了当conda环境出现问题时如何重新安装Anaconda，并提供了详细步骤。

摘要由CSDN通过智能技术生成

One Lab实验室服务器配置

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

参考链接

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