cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项

cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项

  • 对应版本号
  • 最好用conda指令安装,不要用pip指令安装
  • 其他包的安装

今天在配置pycharm环境的时候,一个小项目需要使用GPU(型号1660),所以需要安装tensorFlow-gpu。但是在安装过程中遇到一些小问题,所以来记录一下。

对应版本号

版本号一定要对应上,由于我下载的是cuda10.1。亲测tensorflow-gpu-1.14.0是可以用的。cuda10.0也是可以的。同时cudnn也应该配置相应的版本。若原来有cpu版本的TensorFlow,安装的时候不需要管。

最好用conda指令安装,不要用pip指令安装

conda install tensorflow-gpu==1.14.0

先对环境用activate进行激活,再用conda指令进行安装。安装用pip指令的话会出现很多奇怪的问题。同时用conda指令安装的过程比较缓慢,需要耐心等待。

其他包的安装

在经过了漫长的等待后,在下满心欢喜地跑程序试一试,结果陷入了疯狂uninstall和install其他包的深渊。其中如果不用keras,那就皆大欢喜,是可以跑的。但是用keras一定要对应上相应的版本,这里直接放上我的环境的图,里面有相应的包的版本。

cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项_第1张图片
cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项_第2张图片
cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项_第3张图片
cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项_第4张图片

你可能感兴趣的:(cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项)