Linux搭建深度学习环境

2023年4月11日下午9:46 • 深度学习

一、独显

二、下载显卡驱动

https://www.nvidia.cn/Download/index.aspx?lang=cn

三、安装显卡驱动

1、卸载原有的驱动

sudo apt-get remove --purge nvidia*

2、禁用nouveau

sudo gedit /etc/modprobe.d/blacklist.conf

在文本最后添加以下内容

blacklist nouveau

options nouveau modeset=0

3、保存退出，执行命令使其生效

sudo update-initramfs -u

4、重启电脑

5、执行命令没有任何输出则说明禁用成功

lsmod | grep nouveau

6、将安装包放到Home里面

7、停掉linux图形界面的服务

sudo service lightdm stop

8、按ctrl+alt+f1切换到纯控制台界面

9、进入Home

cd ~

10、给驱动赋可执行权限

sudo chmod a+x NVIDIA-Linux-xxx.run

11、安装

sudo ./NVIDIA-Linux-xxx.run -no-opengl-files

(提示安装基本上都是accept, yes, 当提示你nvidia-xconfig时，如果有双显卡就选择不安装，如果单显卡就选择安装)

12、检验是否安装成功

nvidia-smi

13、启用图形界面

sudo service lightdm start

14、按ctrl+alt+f7切换回图形界面

三、安装cuda

1、下载cuda

https://developer.nvidia.com/cuda-downloads

2、查看nouveau是否禁用成功

lsmod | grep nouveau

3、给文件赋予可执行权限

sudo chmod a+x cuda_xxxx.run

4、安装cuda

sudo ./cuda_xxx.run --no-opengl-libs

安装步骤

Do you accept the previously read EULA?
accept/decline/quit: accept
 
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62?
(y)es/(n)o/(q)uit: n
 
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
 
Enter Toolkit Location
[ default is /usr/local/cuda-8.0 ]:(直接回车)
 
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
 
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y
 
Enter CUDA Samples Location
[ default is /home/tang]:(直接回车)

5、配置环境变量

export PATH=$PATH:/usr/local/cuda_xxx/bin

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda_xxx/lib64

6、使配置生效

source ~/.bashrc

7、验证CUDA Toolkit

nvcc -V

四、配置cudnn

1、下载cudnn

https://developer.nvidia.com/rdp/form/cudnn-download-survey

2、解压cudnn

cp cudnn_xxx-linux-x64.solitairetheme8 cudnn_xxx-linux-x64.tgz