linux下 pytorch gpu cuda cudnn 安装

1. 首先确认linux 是否安装驱动

因为我已经安装了,执行命令:nvidia-smi

 看到如下信息:

第一个框是驱动版本,第二个框支持的cuda最高版本

2. 确认自己要安装的torch版本和cuda版本

Previous PyTorch Versions | PyTorch   到这个网站自己找

下面这个是我自己要安装的版本,但这一步不装,下面装,只是先确定版本

# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu113/torch_stable.html

3. 下载cuda和cudnn文件

具体如何下载参考这篇博客,只需要把window的对应换成linux即可

window下 torch gpu cuda cudnn 安装_泯灭XzWz的博客-CSDN博客_安装torch-gpu

上面 两个包是我下载好的样子

 4. 安装cuda

在cuda安装包目录下,执行

sudo sh cuda_11.1.1_455.32.00_linux.run

几点注意的:

1.如果你已经安装了驱动,会有如下提示:

Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.

不管他直接选择continue

2. 出现如下的界面时,如过你已经安装驱动,选中光标选中Driver,按一下 enter键,把前面的X去掉,然后选择install 回车安装。

3. 安装好后配置环境变量

vi ~/.bashrc

加入如下配置

export PATH=/usr/local/cuda-11.1/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64

 4. 验证cuda是否安装成功

按下面操作

cd /usr/local/cuda-11.1/samples/cuda-samples/Samples/1_Utilities/deviceQuery

sudo make

sudo ./deviceQuery

输入如下信息则安装成功:

5. 安装cudnn

1.解压 安装包

tar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz

2. 复制并更改权限

$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

6. 安装torch

这一步就安装之前我们确认的torch版本

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

7.验证

成功!

8. 参考

CUDA Installation Guide for Linux

觉得有用,帮忙点个赞呗  >.<

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值