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


今天在配置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安装注意事项
cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项
cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项
cuda10.1+cudnn7.6.0+tensorflow-gpu-1.14.0:tensorFlow-gpu安装注意事项

相关文章:

  • 2022-12-23
  • 2021-08-09
  • 2021-06-14
  • 2021-10-02
  • 2021-12-14
  • 2022-01-19
  • 2021-05-15
  • 2021-09-29
猜你喜欢
  • 2021-09-14
  • 2019-01-14
  • 2021-07-17
  • 2021-05-30
  • 2021-08-01
  • 2021-07-25
  • 2021-09-19
相关资源
相似解决方案