最近跟着导师研究深度学习,DL最重要的是GPU和内存。所以GPU的性能是越高越好,内存最好SSD,SSD也是越大越好,唯恐你的GPU太烂啦,O(∩_∩)O哈哈~,显卡计算能力大于3.0。另一个跑GPU速度很快,cuDNN更快,所以得用GPU。再补一句话:不要用虚拟机,听说虚拟机不能用GPU,有精力的可以跳进坑里试试。
说说我的版本:
1 Linux系统:Ubuntu14.04,博主看了Ubuntu16的很多人说G++编译器又出问题,而且我们实验室已经有人掉坑里了,搞了半个月才搞定。补一下:我是64位系统,现在我想应该很少人用32 的吧,否则下面的建议自己找32位的资料包。
2 驱动:NVIDIA-Linux-X86_64-384.90.run,这就得说一下,这个驱动得去英伟达官网,把你显卡型号啊,版本啊告诉它,英伟达会自动帮你选好合适的版本,建议不要随便看别人的版本。说一下我的显卡是GTX1070的,各位一样的话,就可以去网上找这个驱动,因为外网嘛那个速度你懂得。
3 CUDA:cuda-repo-ubuntu1604-8-0-local-8.0.44-1_amd64.deb,这个我看一些教程用的GTX1080显卡也是这个版本,我开始也是cuda的7.5的,不知道什么原因反正跑不起来,后面换成这个可以。
4 cuDNN:cudnn-8.0-linux-x64