欢迎访问我的网站,希望内容对您有用,感兴趣的可以加入免费知识星球。

ubuntu安装CUDA

人工智能 迷途小书童 4年前 (2019-12-13) 11649次浏览 0个评论

软硬件环境

  • 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等(每个版本不太一样)

ubuntu cuda

默认情况下,cuda被安装在了/usr/local下,同时还创建了一个软链接cuda

ubuntu 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

ubuntu cuda

最后执行nvidia-smi看看

ubuntu cuda

刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作

多cuda版本共存及切换

cuda的安装就可以看到,多个版本的cuda共存是可行的,而且切换也非常简单。

安装过程跟上面提到的是一样的,有几个版本就安装几个,比如我说这里安装了2个,一个10.0,一个10.1,且当前激活的是10.1版本

ubuntu cuda

现在我们来操作下切换

# 删除软链接
sudo unlink /usr/local/cuda

# 重新创建软链接
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

参考资料

喜欢 (0)

您必须 登录 才能发表评论!