ubuntu18.04+pytorch配置gpu环境

1.安装GPU显卡驱动      

         首先，确认你是有GPU显卡的：lspci | grep -i nvidia

         检测你的NVIDIA显卡型号和安装推荐的驱动安装型号：ubuntu-drivers devices

带有recommend的为系统推荐的驱动（已用红点标识标识）

        安装推荐驱动：sudo apt-get install nvidia-driver-390

也可以选择自动安装合适的驱动：sudo ubuntu-drivers autoinstall

安装完成后重启：sudo reboot

重启完成后检查是否安装成功：nvidia-smi

 显示上图，表示安装成功。

2.安装CUDA  ToolKit

        首先查看自己的pytorch和cuda版本，进入conda环境后：python

然后导入torch模块：import  torch

打印出pytorch和cuda版本：print(torch.__version__)（注意此处version两侧为双" _ "， 输不对就直接复制命令 ）

结果如上图，pytorch版本为1.12.0、cuda版本为10.2。

        从cuda官网（没有VPN应该也是可以打开的，不行就多试几次）选择对应版本的CUDA  ToolKit进行下载。

 跳出界面，选择ubuntu18.04系统和安装类型(我安装的是deb[loacl]，也有人安装runfile的是版本，博主也不清楚差别在哪里-_-)，然后会自动生成安装指令，如下图：

直接复制命令运行就好(过程中出现问题可以在评论区留下，看到会进行回复)。

        上述完成后，进行环境变量配置：

        打开全局配置文件：sudo vi /etc/profile

        在文件末尾添加如下内容(注意修改cuda版本号，我的是cuda-10.2)：

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64\S${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

        激活环境变量：source /etc/profile
        运行 nvcc -V  查看是否安装成功，结果如下：

 

 3.安装cuDNN

        从官网选择对应版本的安装包进行下载(下载前应该要先注册，自己就可以注册)，如下：

         下载完成后进行安装：

sudo dpkg -i cudnn-local-repo-ubuntu1804-8.4.1.50_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu1804-8.4.1.50/cudnn-local-E3EC4A60-keyring.gpg /usr/share/keyrings/
cd /var/cudnn-local-repo-ubuntu2004-8.4.1.50/
sudo dpkg -i libcudnn8_8.4.1.50-1+cuda11.6_amd64.deb
sudo dpkg -i libcudnn8-dev_8.4.1.50-1+cuda11.6_amd64.deb
sudo dpkg -i libcudnn8-samples_8.4.1.50-1+cuda11.6_amd64.deb

上述命令要根据你的下载包名称进行修改，直接复制运行大概率是要报错的！

        完成后重启。

        重启后检验是否安装成功：

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd  $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN

        出现

 表示安装成功！

此步骤如果出现以下问题

```bash
test.c:1:10: fatal error: FreeImage.h: No such file or directory
    1 | #include "FreeImage.h"
      |          ^~~~~~~~~~~~~
compilation terminated.
>>> WARNING - FreeImage is not set up correctly. Please ensure FreeImage is set up correctly. <<<

运行：sudo apt-get -y install  libfreeimage3 libfreeimage-dev

然后重新检验。