虽然说cutorch已经过时了,但是有的时候还是有需要(其实就是跑别人过去发表论文用的源码而已)。这里将流程简略走一遍(因为是很久以前的事情了),然后将一些其他博文上没提到的问题解决一下。
请优先参考这篇博文,写的很细致(除了标题里面的100%值得商榷——CUDA10.0有一些新的问题):https://blog.csdn.net/zouyu1746430162/article/details/79510773
准备工作
- 对于笔者这样因为要安装cutorch才安装Ubuntu系统的萌新来说,可能还会遇到惯例的开机时显卡驱动导致卡死问题(N卡似乎都是这个样子?),一般解决方案是禁用nouveau,具体操作百度可得。
- 两个版本的gcc。除了最新版的以外,还需要一个4.8版本的应对一些安装中的需要(一些安装包安装过程中,gcc版本过高会有警告)。当然,不排除用高版本编译没有问题的可能,不过笔者是用了两个版本。建议百度切换gcc版本的方法,不用换一个版本重装一次。
- N卡驱动。最好的办法是查好自己的显卡型号,到官网上去查询、下载run文件。或者在终端查看Ubuntu推荐的驱动版本,用apt-get下载。
- CUDA10.0与9.0。为什么会需要9.0呢?因为cutorch的安装调用了CUDA的一个lib文件,但是这个文件到10.0开始就没有了。所以需要将9.0也安装一遍,把这个文件拷贝到10.0的lib目录下,这样就解决了问题。两个安装包都在官网下载。另外,