ubuntu16.04安装CUDA、cuDNN GTX 1660Ti

在安装之前需要先确定已经安装好与显卡适配的驱动,如果没有的话:

  • 下载驱动:https://www.nvidia.cn/Download/index.aspx?lang=cn
  • 安装:https://blog.csdn.net/frothmoon/article/details/88875020

如果安装好了就可以下面的安装了。

安装CUDA

下载

有一些注意事项可以在安装前看一看https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
其中有一点需要注意的,安装的CUDA版本需要和驱动版本对应。
在这里插入图片描述
此外CUDA可以安装多个版本,根据需要可以进行安装,在使用时注意切换版本。

此处下载与驱动版本匹配的CUDA:https://developer.nvidia.com/cuda-toolkit-archive
在页面中根据自己的实际情况选择不同的系统配置,下载即可。
在这里插入图片描述
我安装的是CUDA10.0。

安装

下载后是一个run文件,在下载目录运行sudo sh cuda_xxxxx_linux.run,此处xxxxx替换为你的型号。

  • 一直按空格键到后面,提示接受条款,输入accept接受。
  • 然后询问是否安装驱动,因为前面已经安装过了,输入n不安装。
  • 接着输入y同意安装cuda工具。
  • 输入y安装指向/usr/local/cuda的符号链接。
  • 输入y安装samples,可以测试是否安装成功。
  • 回车确认samples安装的默认路径,/home/用户名,安装路径测试完可以删除。

添加环境变量

命令sudo gedit /etc/profile打开一个文件,在最后添加:

export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH

没有gedit可以用sudo apt-get install gedit安装。
输入的两行中cuda版本要根据你下载的替换,如cuda-9.0等
保存文件后重启系统。

测试

重启后可以测试CUDA Toolkit安装是否正确:nvcc --version,输出信息说明安装正确。

接下来编译samples。
默认路径为~/NVIDIA_CUDA-10.0Samples,命令行cd进入后make,这里需要等一段时间。
之后会生成可执行文件在~/NVIDIA_CUDA-10.0_Samples/bin/x84_64/linux/release,进入目录后输入./deviceQuery
会有一堆输出。
这就安装好啦。

安装cuDNN

cuDNN的安装非常简单,但是下载需要注册账号登录。
官网下载:https://developer.nvidia.com/rdp/cudnn-archive,下载时注意选择与CUDA适配的版本,点开后直接下载for Linux的就可以。
这里用windows下载的文件格式和ubuntu下载的文件格式不一样,但windows下载的solitairetheme8格式文件可以重命名为和ubuntu一样的tar.gz文件。
在这里插入图片描述
我安装的是7.6.5版本。
安装时用tar -xvzf cudnn-10.0-linux-x64-vxxxx.tar.gz进行解压,其中xxxx替换为你的版本。
然后进行一些文件的复制和权限授予。

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 /usr/local/cuda/lib64/libcudnn*

安装完成。

参考教程:http://ttshun.com/2018/09/08/Caffe%E7%9A%84GPU%E6%A8%A1%E5%BC%8F%E5%AE%89%E8%A3%85/

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值