前提
关于如何查看显卡型号、显卡支持的cuda版本,以及对应的tensorflow-gpu版本等的方法网上可以很容易查到,不再赘述。
安装环境为(注:英韦达官网上安装cuda9.0时,没有19.10选项,我选择的是17.04):
- 显卡:GeForce940MX
cuda
: 9.0cudnn
: 7.0.5tensorflow-gpu
: 1.8.0
经过测试,可以实现显卡加速运算。
安装过程
安装过程主要参考了:
2)Ubuntu 18.04安装Tensorflow(GPU)
注意:只是参考,对应版本号需要相应更改。
注意事项
1)在安装cuda时,直接执行:sudo sh cuda_9.0.176_384.81_linux.run
会报错并按安装失败,我的解决方法是:
- 首先
sudo apt install nvidia-cuda-toolkit
- 然后:
sudo sh cuda_9.0.176_384.81_linux.run --override
2)安装完tensorflow后,最好重启一下,或者至少在安装cudnn后要重启一下再进行接下来的步骤,不然可能会在安装完tensorflow后测试失败。
转载:https://blog.csdn.net/WUDIxi/article/details/104538268
查看评论