Skip to content

Latest commit

 

History

History
191 lines (124 loc) · 4.59 KB

CUDA.md

File metadata and controls

191 lines (124 loc) · 4.59 KB

安装CUDA和CUDNN

安装NVIDIA驱动

可以参考：Installation/NVIDIA.md

安装CUDA

  • 下载完CUDA 9.0之后执行如下语句，运行.run文件
sudo sh cuda_9.0.176_384.81_linux.run

单击回车，一路往下运行。

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n  # 选择否，因为已经安装好驱动程序了

Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
 [ default is /usr/local/cuda-9.0 ]:  # 记住安装位置， 默认是安装在/usr/local/cuda文件夹下

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: n  # 根据需要选择y或者n

Install the CUDA 9.0 Samples?
(y)es/(n)o/(q)uit: y

Enter CUDA Samples Location
 [ default is /home/ubuntun ]: 

Installing the CUDA Toolkit in /usr/local/cuda-9.0 ...
Missing recommended library: libGLU.so
Missing recommended library: libXmu.so

Installing the CUDA Samples in /home/ubuntun ...
Copying samples to /home/ubuntun/NVIDIA_CUDA-9.0_Samples now...
Finished copying samples.
  • 配置环境变量，方法一（所有用户，建议使用方法二）：

运行如下命令打开profile文件

sudo gedit /etc/profile

打开文件后在文件末尾添加路径，也就是安装目录，命令如下：

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

保存，然后重启电脑:

sudo reboot
  • 配置环境变量，方法二（当前用户，建议使用该方法）：

运行如下命令打开.bashrc文件

sudo gedit ~/.bashrc

打开文件后在文件末尾添加路径，也就是安装目录，命令如下：

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

保存，然后使之生效:

source ~/.bashrc
  • 测试安装是否成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

如果显示关于GPU的信息就是成功了。

  • 多个CUDA之间切换
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-x.x /usr/local/cuda

安装cudnn

Inference: Install_AND_path/install nvidiaDriver_cuda_cudnn.md

  1. 下载对应版本的cudnn

  2. 需要下载两个版本，runtime版和developer版

cuDNN Runtime Library for Ubuntu18.04 (Deb)
cuDNN Developer Library for Ubuntu18.04 (Deb)

  1. 将下载的文件格式改为.tgz，然后解压

  2. 一个里版本里面有头文件(在include中)，一个里面有动态连接文件（在lib中）

  • 将include中头文件cudnn.h拷贝到/usr/local/cuda/include
  • 将lib中的动态连接文件拷贝到/usr/local/cuda/lib64
  1. 添加环境变量

如果添加过环境变量，就可以不添加。可以通过echo $LD_LIBRARY_PATH查看是否添加。

运行如下命令打开.bashrc文件

sudo gedit ~/.bashrc

打开文件后在文件末尾添加路径，也就是安装目录，命令如下：

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

保存，然后使之生效:

source ~/.bashrc
  1. 对拷贝进来的两个动态连接库创建软链接
cd /usr/local/cuda/lib64/
sudo ln -s libcudnn.so.7 libcudnn.so
# sudo ln -s libcudnn.so.7.x.x libcudnn.so.7

问题一：Pycharm中找不到cuda

Inference: Install_AND_path/无法找到cuda9.0

在添加cuda环境变量后，一直无法找到cuda9.0:

  1. 有可能是在sudo下执行的，需要在sudo下添加环境变量

  2. 添加lib

打开文件cuda.conf：

sudo gedit /etc/ld.so.conf.d/cuda.conf

添加如下内容并保存：

/usr/local/cuda/lib64 
/lib
/usr/lib
/usr/lib32

问题二：ImportError: libcublas.so.x.x: cannot open shared object file: No such file or directory

Inference: ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

  • 确保安装了对应版本的CUDA：否则，回到安装CUDA的步骤

  • 确保路经在PATH环境变量中(echo $PATH)：否则，回到安装配置环境变量的步骤

  • 如果还不行，检查/usr/local/cuda-x.x/lib64下是否有libcublas.so.x.x：如果有，则sudo ldconfig /usr/local/cuda-x.x/lib64