本机环境
Win10,1660Ti,VS2017,Python3.7.6,CUDA10.0.130,cuDNN7.6.5,Tensorflow-gpu1.14.0
安装顺序
显卡驱动(笔记本官网驱动)→64位Anaconda3→VS2017→创建虚拟环境,并安装Tensorflow-gpu1.14.0第三方库→安装CUDA10.0和cuDNN7.6→测试
安装注意事项
- 主要参考安装教程,请先阅读注意事项,再看安装教程。
- CUDA的版本选择很重要。NVIDA的控制面板中,说明1660Ti是支持CUDA10.1的,但是在我安装完CUDA10.1后,运行 import tensorflow 时出现 DLL 加载失败,我不得不选用CUDA10.0。后来在参考别人的博客时,发现:
① 有博主推荐,安装的CUDA的版本要比可支持的版本低一点;
② 也有博主1660Ti安装CUDA10.1成功,但是VS的版本是2019。
但是我均没有验证过,考虑到有的项目用2017比较多,而且卸载麻烦,所以没有尝试 - 安装Tensorflow-gpu1.14.0第三方库网速慢的问题。网上找到whl文件挺麻烦,所以通过镜像下载即可:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0
- tensorflow DLL加载失败。有很多博文的解决方法都可以尝试,但是我觉得最后实在不行可以重装CUDA,版本问题可能比较常见。
- CUDA卸载。当然最好一次性成功,如果需要卸载的话,可以在控制面板中按时间排个序,把当天的有关NVIDA的全部卸载了,之前的显卡驱动相关的保留,如NVIDA图形驱动,PhysX等。
- 使用pycharm测试。我参考的博客是用cmd进行测试的,也可以在pycharm里测试一遍,只需要将pycharm中的project interpreter更改成自己虚拟环境中Scripts下的python.exe即可。
其他
- 我还未安装tensorflow2.0,有待后面测试
- 文章如有错误,请指正
- 安装中如有疑问或者找不到安装包的,欢迎留言。