当前位置：首页 > news >正文

linux系统pytorch重装

news 来源：原创 2024/9/20 19:46:48

""
一些查看设备的命令
""
#导出导入环境
conda env export > env.yaml 
conda env create -n [name] -f env.yaml

# 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
#cpu内存
free -h

nvidia-smi 可以看到GPU使用情况以及总内存
在这里插入图片描述

问题描述：在运行一个程序的时候，发现服务器的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选项，其实也没有很慢。了解所用的系统是配置环境的必要条件！
下载再安装 https://download.pytorch.org/whl/torch_stable.html pip install xxxx.whl 安装前要先卸载现有的。
https://zhuanlan.zhihu.com/p/354177054#:~:text=%E4%BA%94%E3%80%81%E5%AE%89%E8%A3%85pytorch%201%E3%80%81%E6%89%93%E5%BC%80PyCharm%EF%BC%8C%E8%BF%9B%E5%85%A5%E5%88%9A%E5%88%9A%E5%88%9B%E5%BB%BA%E7%9A%84%E5%B7%A5%E7%A8%8B%E9%87%8C%EF%BC%8C%E6%89%93%E5%BC%80Terminal%E7%AA%97%E5%8F%A3%EF%BC%9A,2%E3%80%81%E5%9C%A8PyCharm%E7%9A%84Terminal%E7%AA%97%E5%8F%A3%E4%B8%AD%E9%94%AE%E5%85%A5cd%20%E5%91%BD%E4%BB%A4%EF%BC%8C%E5%88%87%E6%8D%A2%E7%9B%AE%E5%BD%95%E5%88%B0PyTorch%E5%AE%89%E8%A3%85%E5%8C%85%E6%89%80%E5%9C%A8%E7%9A%84%E5%9C%B0%E5%9D%80%EF%BC%8C%E7%84%B6%E5%90%8E%E5%88%86%E5%88%AB%E6%89%A7%E8%A1%8Cpip%20install%E5%91%BD%E4%BB%A4%E5%AE%89%E8%A3%85%E4%B8%A4%E4%B8%AA%E6%96%87%E4%BB%B6%EF%BC%8C%E5%85%88%E5%AE%89%E8%A3%85torch%EF%BC%8C%E5%86%8D%E5%AE%89%E8%A3%85torchvision%EF%BC%8C%E6%9C%9F%E9%97%B4%E4%BC%9A%E4%B8%80%E8%B5%B7%E4%B8%8B%E8%BD%BD%E6%89%80%E9%9C%80%E8%A6%81%E7%9A%84%E4%BE%9D%E8%B5%96%E5%8C%85