当前位置:   article > 正文

RTX3090 GPU环境配置_3090显卡设置

3090显卡设置

前言

这篇文章中安装的版本都是根据我自己的电脑的情况选择的,有需要的可以根据实际情况选择,
我的系统上Ubuntu20.04,显卡是RTX3090。跟我一样的话就大胆copy,不一样的话就注意点,名字上会有区别。

安装Nvidia显卡驱动

nvidia-smi
  • 1

方法1:
打开Software & Updates > Additional Drivers选择第一个Using NVIDIA driver metapackage from nvidia-driver-460(proprietary,tested)然后Apply Changes,就开始下载驱动了。
在这里插入图片描述安装完成是这样的:
在这里插入图片描述


方法2:
或者直接到官网下载下来

驱动程序下载
在这里插入图片描述
下载对应驱动,我这里的驱动是NVIDIA-Linux-x86_64-460.56.run

chmod +x NVIDIA-Linux-x86_64-460.56.run
sudo sh NVIDIA-Linux-x86_64-460.56.run
  • 1
  • 2

一路确定OK


装完驱动之后,重启电脑:
在这里插入图片描述
这样驱动就安装成功了
可以更新一下软件:

sudo apt update
sudo apt upgrade
  • 1
  • 2

安装CUDA

nvcc -V
# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2020 NVIDIA Corporation
# Built on Mon_Oct_12_20:09:46_PDT_2020
# Cuda compilation tools, release 11.1, V11.1.105
# Build cuda_11.1.TC455_06.29190527_0
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

这是已经装好了CUDA11.1,但是这里的版本和上面的截图里CUDA Version: 11.2不太一样

CUDA Toolkit Archive
CUDA Toolkit 11.2
在这里插入图片描述

wget https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run
chmod +x cuda_11.2.1_460.32.03_linux.run
sudo sh cuda_11.2.1_460.32.03_linux.run
  • 1
  • 2
  • 3

在这里插入图片描述因为前面已经可以看到,我之前是已经安装了CUDA的,这里建议先删除在安装。

我这里就把所有东西都给删了,结果发现刚装的驱动也没了。又走了一遍安装驱动。所以这一步建议在最开始就删除。

sudo apt-get remove --purge nvidia*
  • 1

可是还是一样,那就不管了,直接Continue
在这里插入图片描述输入accept
在这里插入图片描述第一项按空格键反选,因为驱动已经安装过了,版本也一致。
然后移动到最后一行的Install
在这里插入图片描述选择yes,然后根据上图提示需要配置环境变量:

gedit ~/.bashrc
  • 1

在文件最后加入以下语句:

# 这里的cuda路径不加版本号,以便后续升级cuda版本的时候不需要重新配置
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
  • 1
  • 2
  • 3
  • 4

然后使其生效:

source ~/.bashrc
  • 1

安装的CUDA版本统一被安装到/usr/local文件夹下:

ls -l /usr/local
  • 1

系统是通过/usr/local/下的cuda符号链接来识别CUDA当前指定的CUDA版本
在这里插入图片描述
可以看到路径下有cuda-11.1cuda-11.2两个版本,但是使用的是哪个呢?就看上面这个cuda指向哪里cuda -> /usr/local/cuda-11.2/

cd /usr/local/
# 删除原来的软链接
sudo rm cuda
# 新建新的软链接
sudo ln -s /usr/local/cuda-11.2 /usr/local/cuda
  • 1
  • 2
  • 3
  • 4
  • 5

查看版本:

nvcc -V
# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2021 NVIDIA Corporation
# Built on Thu_Jan_28_19:32:09_PST_2021
# Cuda compilation tools, release 11.2, V11.2.142
# Build cuda_11.2.r11.2/compiler.29558016_0
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

安装cuDNN

ls /usr/local/cuda/include/cudnn.h |grep CUDNN_MAJOR -A 2
  • 1

cuDNN Download
下载cuDNN需要注册账号
在这里插入图片描述
选择适合CUDA11.2的第一项,在它下面选择cuDNN Library for Linux (x86_64)下载:
在这里插入图片描述
这是个.tgz的压缩包。直接解压就可以用了。不过这个包还挺大的,压缩包就有1.2G。

下载完成后将其解压,命令如下:

# -C解压到指定文件夹
tar -xzvf cudnn-11.2-linux-x64-v8.1.1.33.tgz -C cudnn-112
  • 1
  • 2

使用以下两条命令复制这些文件到cuda目录下/usr/local/cuda

cd cudnn-112
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/
  • 1
  • 2
  • 3

copy完成之后,查看cuDNN的版本信息:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
  • 1

在这里插入图片描述这样就全部安装完毕了。


  1. 安装驱动可以不独立安装,在安装cuda时一起安装
  2. 安装cuDNN其实就是将一个压缩包中的文件放到cuda安装路径的对应位置
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/寸_铁/article/detail/899637
推荐阅读
相关标签
  

闽ICP备14008679号