Ubuntu 安装CUDA
Ubuntu 安装CUDA_ubuntu安装cuda_Cynthia-石头的博客-CSDN博客
很多刚接触深度学习的朋友面临的第一个大坎就是开发环境的配置,经常一折腾就是一两天,更有甚至十天半月也不一定能让代码跑起来,实在是大大滴不友好。
环境配置问题一般都是各个框架与cuda版本的适配问题,装不好cuda就没办法调用gpu进行快速训练或推理,所以深度学习要是用不上gpu这不相当于学习了个寂寞!多提一句,很多童鞋一开始就是接触的tensorflow,特别在tf1.x时代,各个tf版本与不同cuda版本的适配,经常被搞得欲仙欲死,直接从入门到放弃。
今天给大家简单介绍一下如何在Ubuntu系统安装多个版本的cuda,有需要时顺滑进行切换,非常方便。
cuda、cudnn官方下载链接:
CUDA Toolkit Archive | NVIDIA Developer
在官网下载不同版本的cuda版本以及对应的cudnn版本,cudnn需要先注册一下。
这里最好下载runfile的文件格式
下载好的文件如下,这里以cuda10.1为例:
运行以下命令开始安装
sudo chmod +x cuda_10.1.105_418.39_linux.run
sudo ./cuda_10.1.105_418.39_linux.run
1、选择accept
2、按回车选择不安装驱动,然后选择Install进行安装
3、选择默认安装路径
# 设置当前用户的环境变量
sudo gedit ~/.bashrc
# 在.bashrc文件后面添加如下内容,注意这里不指定具体的cuda版本,就是为了方便切换
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
# 终端运行,使之生效
source ~/.bashrc
检查安装是否成功,终端输入ncvv --version
这里的cuda已经指向刚安装的cuda10.1的版本,是因为刚安装的时候默认选择了。
安装cuda10.1对应的cudnn,如果目前的cuda软连接指向的是10.1,直接复制过去即可,步骤如下:
tar -zcvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
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
至此,cuda10.1已经安装完成了。假设该机子上已经安装了cuda10.0的版本,如何切换回去呢?
也非常简单。
# 删除目前的软链接
sudo rm -rf /usr/local/cuda
# 生成新的软链接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
# 查看当前cuda的版本,或者用nvcc --version
cat /usr/local/cuda/version.txt
参考:
Ubuntu 18.10 下安装CUDA10/CUDA10.1_cuda 10.1 ubuntu 18_哈里哈气的博客-CSDN博客
CUDA10.0和CUDA10.1同时安装,Linux系统下如何切换_yimaoshu的博客-CSDN博客
Ubuntn16.04下cuda10.0和cuda9.0自由切换_安装了cuda10.0但是版本还是9_kelly1250230225的博客-CSDN博客
Ubuntu16.04下成功安装多版本的Cuda和切换Cuda环境 --Cuda9.0 & Cuda10.0_cuda9.0和cuda10.0_梦坠凡尘的博客-CSDN博客
转载自:深度觉醒 公众号
欢迎大家扫码关注!