这是我参与8月更文挑战的第7天,活动详情查看:8月更文挑战
软硬件环境
- ubuntu 18.04 64bit
- NVidia GTX 1070
- CUDA 10.1
- cudnn 7.6.3
安装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,官方下载地址: developer.nvidia.com/cuda-downlo…
# 添加可执行的权限 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
这样几个重要的环境变量就立马生效了。或者,重新打开terminal,也同样生效。
安装cudnn
到官网 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 # 重新创建软链接,以使用10.1为例 sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
参考资料
- www.nvidia.com/Download/in…
- developer.nvidia.com/cuda-toolki…
版权声明:
本文来源网络,所有图片文章版权属于原作者,如有侵权,联系删除。
本文网址:https://www.mushiming.com/mjyfx/16212.html