软硬件环境
- ubuntu 18.04 64bit
- NVidia GTX 1070
- CUDA 10.1
- cudnn 7.6.3
视频看这里
此处是youtube
的播放链接,需要科学上网。喜欢我的视频,请记得订阅我的频道,打开旁边的小铃铛,点赞并分享,感谢您的支持。
安装nvidia驱动
进入没有安装驱动的ubuntu
系统,屏幕显示是不正常的,分辨率偏小。从官网下载RUN
文件,比如我这里的NVIDIA-Linux-x86_64-435.21.run
,这里的版本选择需要与后面的cuda
进行匹配
# 添加执行权限
chmod a+x NVIDIA-Linux-x86_64-435.21.run
sudo ./NVIDIA-Linux-x86_64-435.21.run
安装过程基本上就是选择yes
或者no
,装完后重启系统,就会发现屏幕显示正常了,分辨率也变高了。
由于cuda
的安装文件中已经包含了对应的驱动,所以,这一步,其实可以不用先做,当然了,这样安装也没有问题,还是看个人习惯
安装CUDA
接下来开始安装cuda
,官方下载地址: https://developer.nvidia.com/cuda-downloads
# 添加可执行的权限
chmod +x cuda_10.1.243_418.87.00_linux.run
sudo ./cuda_10.1.243_418.87.00_linux.run
安装过程中会有一些选项出现,根据自己的实际情况选择,比如要不要使用cuda
自带的驱动,安装路径、是否安装samples
等(每个版本不太一样)
默认情况下,cuda
被安装在了/usr/local
下,同时还创建了一个软链接cuda
然后配置下环境变量,编辑~/.bashrc
文件
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
最后执行下source
source ~/.bashrc
这样几个重要的环境变量就立马生效了
安装cudnn
到官网 https://developer.nvidia.com/cudnn 去下载,这里需要注册,有点麻烦。需要注意的是,cudnn
的版本是要跟cuda
匹配的
tar xvf cudnn-10.1-linux-x64-v7.6.3.36.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda-10.1/include
sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
# 添加读的权限
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
测试
安装完毕,我们来测试下环境是否都ok
?
进入到sample
中,跑下deviceQuery
# 安装编译工具
sudo apt install build-essential
cd NVIDIA_CUDA-10.1_Samples/1_Utilities/deviceQuery
make
最后执行nvidia-smi
看看
刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作
多cuda版本共存及切换
从cuda
的安装就可以看到,多个版本的cuda
共存是可行的,而且切换也非常简单。
安装过程跟上面提到的是一样的,有几个版本就安装几个,比如我说这里安装了2个,一个10.0,一个10.1,且当前激活的是10.1版本
现在我们来操作下切换
# 删除软链接
sudo unlink /usr/local/cuda
# 重新创建软链接
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda