1.cuda11.0的配置

查看當前顯卡驅動版本：nvidia-smi

driver version.png

可以看出當前cuda驅動最高支持cuda11.0,即不用更新顯卡驅動，450.102.04足夠了

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run --no-check-certificate

在當前目錄，可以看到cuda_11.0.2_450.51.05_linux.run

cuda11.0.png

sudo chmod 775 cuda_11.0.2_450.51.05_linux.run
sudo ./cuda_11.0.2_450.51.05_linux.run

step1.png

step2.png

step3.png

step4.png

step5.png

cuda11.0安裝路徑.png

cuda安裝路徑.png

2.cudnn8.1.1配置

https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.1.1.33/11.2_20210301/cudnn-11.2-linux-x64-v8.1.1.33.tgz

win10下載好了再拉到linux環(huán)境后綴改名為tgz,再解壓

cudnn8.1.1.png

解壓后文件名為cuda,

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64/
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

設置環(huán)境變量（編輯文件.bashrc）

export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

創(chuàng)建cuda.sh腳本，用來快速更換cuda版本，只需更改以下所示的cuda版本安裝文件夾名字即可

#! /bin/bash # employ bash shell
if [ -d "/usr/local/cuda" ];then
  echo "cuda文件夾存在"
  sudo rm -r /usr/local/cuda
else
  echo "將重新建立軟連接。"
fi

sudo ln -s /usr/local/cuda-11.0 /usr/local/cuda 
echo "現(xiàn)將cuda更換為:"
nvcc -V
echo "現(xiàn)將cudnn更換為:"
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

sudo sh cuda.sh
nvcc --version

cuda11.0打印信息.png

tensorflow-gpu==2.4.1配置

conda create -n tf2.4 python=3.7
conda activate tf2.4
pip install tensorflow-gpu==2.4.1

print(tf.__version__)
print( tf.test.is_gpu_available())

打印版本信息.png

cuda是否可用.png