linux非root用户/个人用户更新CUDA和cudnn并安装anaconda和pytorch配置环境

宇宙KU

已于 2024-08-01 11:18:16 修改

阅读量243

点赞数 3

文章标签： linux 服务器 pytorch 人工智能

于 2024-07-31 19:13:03 首次发布

本文链接：https://blog.csdn.net/Univeir/article/details/140829853

版权

（入门深度学习，cuda版本太旧不能安装新的pytorch，查询多种方法，数次尝试多个版本的cuda python pytorch 安装失败，最后基于前辈们的智慧总结出了成功方法，希望能为你节省时间）

1. 查询服务器driver version

在终端输入nvidia-smi ，示例：

对应nvidia官网给的文件CUDA 12.5 Update 1 Release Notes (nvidia.com)

找到可用的cuda版本，如图，示例可用的最高版本为cuda 11.8

2. 安装cuda及cudnn

参考：linux非root用户安装CUDA和cudnn(centos7 cuda12.3为例)_centos7安装cuda12-CSDN博客

3. 安装anaconda

（到参考文献的1.3步，即anaconda安装成功）

4. 建立和配置环境（python和pytorch）

首先存在版本对应问题

python和pytorch有对应问题，参考：版本匹配指南：PyTorch版本、Python版本和pytorch_lightning版本的对应关系_pytorch-lightning版本对应-CSDN博客

版本匹配指南：PyTorch版本、Python版本和pytorch_lightning版本的对应关系_pytorch-lightning版本对应-CSDN博客

注意：pytorch与cuda有版本对应问题（如下图所示），pytorch版本选取可以步骤1升级后的cuda版本为准，即在终端输入 nvcc -V 获得的版本（此时nvcc -V 和 nvidia-smi 获得的版本可能不一致）

pytorch的历史版本：Previous PyTorch Versions | PyTorch

通过红框信息选择版本，蓝框内是安装的代码：

在终端输入 conda create -n 环境名 python=

示例安装python3.11.0 和 pytorch 2.3.1

以此在终端实行（遇见proceed(y/n)输入y并回车）

conda create -n pytorch2.3 python=3.11.0

conda activate pytorch2.3

将上述pytorch蓝框安装信息复制，在终端输入：

conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia

具体步骤可参考：远程服务器配置 Anaconda 并安装 PyTorch 详细教程_远程服务器安装anaconda-CSDN博客

最后检验：

(base): conda activate pytorch2.3
(pytorch2.3): python3.11 # 进入python环境 注意python版本是否对应
>>> import torch   # 输入并按回车,不报错即pytorch安装成功
>>> torch.__version__  # 查看 PyTorch 版本号
>>> torch.cuda.is_available()  # 查看 CUDA 即 GPU 是否可用
True  # 如果这里是ture那么恭喜你可以用了 如果是false那么看一下哪里出错
>>> exit()

注： driver version、cuda、cudnn、python、pytorch版本均有对应问题，建议在安装前先查询driver version版本，找到对应可安装的cuda版本范围并在pytorch官网查询想安装的pytorch版本是否有合适的安装包，再进行上述所有安装操作。