Ubuntu20.04 CUDA+CUDNN

======================================【Ubuntu基础配置】========================================= 

显卡：RTX4090 *2张 

系统：Ubuntu 20.04 LTS 

Python版本：Python 3.8.10 

  

  

1、更新系统apt数据源 

sudo vim /etc/apt/sources.list  

  

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse  

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse  

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse  

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse  

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse  

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse  

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse  

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse  

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse  

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse  

  

更新源 

sudo apt-get update  

  

2、安装ssh、ifcnofnig、screen、htop、nvidia-cuda-toolkit 

sudo apt install openssh-server       SSH服务 

sudo apt install net-tools            网络工具包，如ifconfig 

sudo apt install nvidia-cuda-toolkit  CUDA工具包 

sudo apt install screen               多用户会话 

sudo apt install htop                 系统监测，如CPU、内存 

  

3、安装pip3 

sudo apt install python3-pip 

设置为国内源 

pip3 install pip -U 

pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple 

  

4、安装torch CUDA 11.8 

下载地址：https://pytorch.org/get-started/locally/ 

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 

  

5、安装其他依赖库 

执行命令 

pip3 install -r pip.txt 

  

gitpython 

ipython  # interactive notebook 

matplotlib>=3.2.2 

numpy>=1.18.5 

opencv-python>=4.1.1 

Pillow>=7.1.2 

psutil  # system resources 

PyYAML>=5.3.1 

requests>=2.23.0 

scipy>=1.4.1 

thop>=0.1.1  # FLOPs computation 

tqdm>=4.64.0 

tensorboard>=2.4.1 

pandas>=1.1.4 

seaborn>=0.11.0 

easydict 

gdown 

lap 

filterpy 

onnx>=1.9.0 

export 

tensorrt 

pycuda 

jwt 

python-multipart 

uvicorn 

  

======================================【安装驱动、CUDA、CUDNN、Tensort】========================================= 

  

1、安装RTX4090驱动 

下载地址：https://www.nvidia.com/download/driverResults.aspx/205863/en-us/ 

执行命令安装： 

sudo sh NVIDIA-Linux-x86_64-535.54.03.run 

  

2、安装CUDA 12.2 

下载地址：https://developer.nvidia.com/cuda-downloads 

  

执行命令，下载本地安装包 

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run 

sudo sh cuda_12.2.0_535.54.03_linux.run 

报错：请先切换为命令行，然后重启reboot系统后再执行即可 

sudo systemctl set-default multi-user.target    命令行模式 

sudo systemctl set-default graphical.target     图形化模式 

  

3、安装CUDNN v8.9.2 for CUDA 12.X 

下载地址：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b 

下载DEB安装包 

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.2.26_1.0-1_amd64.deb 

报错：key，则需要先拷贝key到指定目录“cp .... key"，然后再执行安装包 

  

4、安装Tensort TensorRT 8.6 GA for x86_64 

下载地址：https://developer.nvidia.com/nvidia-tensorrt-8x-download 

https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/secure/8.6.1/local_repos/nv-tensorrt-local-repo-ubuntu2204-8.6.1-cuda-12.0_1.0-1_amd64.deb 

下载DEB安装包 

sudo dpkg -i nv-tensorrt-local-repo-ubuntu2004-8.6.1-cuda-12.0_1.0-1_amd64.deb 

报错：key，则需要先拷贝key到指定目录“cp .... key"，然后再执行安装包 

  

    

sudo apt install nvidia-cuda-toolkit 

设置正确的环境变量 

export CPATH=:/usr/local/cuda-12.2/bin:/usr/local/cuda-12.2/include:/usr/local/cuda-12.2/ 

------------------ 

  

安装完毕！！！