ubuntu环境下安装cuda和cudnn

1、安装显卡驱动

最简单的驱动下载方法,在ubunut系统设置-》软件和更新-》附加驱动-》更新nvidia和intel

2、下载cuda和cudnn

下载cudnn,注意下载对应cuda所支持的版本,不然会出现不兼容的问题。
点我下载cuda
点我下载cudnn(需要帐号登录就很烦)

3、安装cuda

安装
设置安装包权限

sudo chmod +x cuda_10.0.130_410.48_linux.run 

直接运行安装

sudo ./cuda_10.0.130_410.48_linux.run 

注意安装过程中有个显卡驱动的选项选择no,其他都可以yes和默认。
安装过程中,如果出错,u依据提示进入log日志查看错误原因,如果驱动和cuda的版本兼容,很大可能gcc版本不支持(一般是gcc版本不能过高),查看该gcc安装链接解决此问题,我安装的版本需要gcc-7版本。

设定环境变量以及路径:
安装后自动会创建安装目录的软链接/usr/local/cuda ,如果没有可以手动创建

sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda

添加路径

sudo gedit ~/.bashrc 

在最后面添加如下内容:

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

更新

source ~/.bashrc 

如果安装的多个cuda版本,然后想相互切换则更换软链接就行了,更换方式如下:

sudo rm -rf cuda
sudo ln -s /usr/local/cuda-9.1 /usr/local/cuda

思路就是先删除当前软链接,重新添加一个其他版本的,如果之前添加过环境变量,那新安装的版本的环境变量都不用添加了。
最后查看cuda软链接是否更改。

stat cuda 		

查看cuda安装是否成功

nvcc --version		

3、安装cudnn

解压安装包

tar -zxvf cudnn-10.0-linux-x64-v7.6.5.32.tgz cuda/

拷贝对应文件放入cuda安装路径下的对应位置

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

修改对应文件的权限

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

查看cudnn是否安装成功

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

设定环境变量以及路径,这里不需要设置了,cudnn头文件和库都是拷贝到、cuda路径下,因为之前cuda安装时已经设置了。

你可能感兴趣的:(深度学习,深度学习,神经网络,tensorflow,pytorch,机器学习)