欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

linux系统pytorch重装

程序员文章站 2022-06-11 22:00:18
...
""
一些查看设备的命令
""
# cuda version
cat /usr/local/cuda/version.txt
# ubuntu infomation
cat /proc/version  
#查看驱动
nvidia-smi
#操作系统版本
cat /etc/lsb-release
#pytorch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
#gpu数量
torch.cuda.device_count()
#返回gpu名字,设备索引默认从0开始
torch.cuda.get_device_name(0)
#返回当前设备索引
torch.cuda.current_device()
#nvida 版本
cat /proc/driver/nvidia/version
#环境中的包
conda list

问题描述:在运行一个程序的时候,发现服务器的pytorch-cuda不可用,因为之前安装的是cpu版本的。
解决方案:1.在这个环境卸载重装pytorch 2.新建环境

卸载pytorch

conda uninstall pytorch 出错了,昨天按照一个人的博客加了一些外部源就一直报错。。

列出channel
conda config --show

删除添加的channel
conda config --remove channels

删除环境
conda remove -n name --all

新建环境
conda create -n name python=3.7

报错 An unexpected error has occurred. Conda has prepared the above report
解决:conda clean --packages && conda clean --all && conda update --all

安装pytorch(cuda 10.2)
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 (镜像源)

报错
An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.
#添加清华镜像源
conda config --add channels ‘https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/’

总结:添加清华镜像源理论上会快一点,但在网友博客中找到的镜像源都不能用(昨天测试时时好用的),最后还是用了-c选项,其实也没有很慢。了解所用的系统是配置环境的必要条件!

相关标签: 一些问题 linux