在Ubuntu18.04上安装CUDA10.0与cutorch步骤简述(无图)

虽然说cutorch已经过时了,但是有的时候还是有需要(其实就是跑别人过去发表论文用的源码而已)。这里将流程简略走一遍(因为是很久以前的事情了),然后将一些其他博文上没提到的问题解决一下。
请优先参考这篇博文,写的很细致(除了标题里面的100%值得商榷——CUDA10.0有一些新的问题):https://blog.csdn.net/zouyu1746430162/article/details/79510773

准备工作

  1. 对于笔者这样因为要安装cutorch才安装Ubuntu系统的萌新来说,可能还会遇到惯例的开机时显卡驱动导致卡死问题(N卡似乎都是这个样子?),一般解决方案是禁用nouveau,具体操作百度可得。
  2. 两个版本的gcc。除了最新版的以外,还需要一个4.8版本的应对一些安装中的需要(一些安装包安装过程中,gcc版本过高会有警告)。当然,不排除用高版本编译没有问题的可能,不过笔者是用了两个版本。建议百度切换gcc版本的方法,不用换一个版本重装一次。
  3. N卡驱动。最好的办法是查好自己的显卡型号,到官网上去查询、下载run文件。或者在终端查看Ubuntu推荐的驱动版本,用apt-get下载。
  4. CUDA10.0与9.0。为什么会需要9.0呢?因为cutorch的安装调用了CUDA的一个lib文件,但是这个文件到10.0开始就没有了。所以需要将9.0也安装一遍,把这个文件拷贝到10.0的lib目录下,这样就解决了问题。两个安装包都在官网下载。另外,CUDA的示例文件中有一个检测CUDA是否运行正常,具体看官网指示。
  5. cudnn,如果是与CUDA10.0对应的版本应该是7.0版本。一定要安装对应CUDA的版本!

简要步骤

  1. 安装显卡驱动。安装显卡驱动前在BIOS页面关闭Security Boot。在登陆界面输入密码的时候按F2, (F3以及其他Fx按钮也行)切换到文本界面,然后开始安装,注意要加上不安装opengl文件的选项。用nvidia-smi命令来检测驱动是否运行正常。
  2. 不要重启,也不要退出文本界面。继续安装CUDA10.0,不要安装提供的N卡“加速驱动”,因为已经安装过了。最后会提示你未完整安装(驱动未安装),属于正常现象。按照官网的说明,跑CUDA的检测示例文件,如果检测到合适的显卡设备就说明运行正常。
  3. 按F1键就可以回到图形登陆界面了。登陆,打开终端,安装cudnn。
  4. 安装torch,一般来说torch的安装包也会带cutorch,不过建议还是按照参考的方案做一次“升级”。安装cutorch的时候会出现找不到一个lib文件的问题……这个问题的解决方案上面提到过。回到文本界面(需要重启),安装CUDA9.0(只要工具包就可以了),然后在工具包里面找缺失的那个文件,复制一份到10.0的对应位置,就可以顺利编译运行了。
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值