江苏和住房建设厅网站,一家专门做动漫的网站,网站布局策划案,做网站湘潭TensorFlow安装踩坑记录第一步#xff0c;切换服务器cuda版本第二步#xff0c;conda安装tensorflow记录一下最近安装Tensorflow v1时遇到的问题和解决办法第一步#xff0c;切换服务器cuda版本
首先我想安装tensorflow 1.13.1#xff0c;兼容的cuda版本是10.0#xff0c…
TensorFlow安装踩坑记录第一步切换服务器cuda版本第二步conda安装tensorflow记录一下最近安装Tensorflow v1时遇到的问题和解决办法第一步切换服务器cuda版本
首先我想安装tensorflow 1.13.1兼容的cuda版本是10.0但我服务器上的cuda版本是11所以研究了一下参考以下两篇博文切换了服务器上cuda runtime版本同时继续保持cuda driver版本为11。 https://www.jianshu.com/p/eb5335708f2a https://cloud.tencent.com/developer/article/2000757 这一步做完后我可以正常import tensorflow并创建placeholder, variable了。
第二步conda安装tensorflow
但是在运行模型的时候tensorflow报错
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize,按照以下这篇排查后最后还是版本兼容问题 https://stackoverflow.com/questions/53698035/failed-to-get-convolution-algorithm-this-is-probably-because-cudnn-failed-to-in
tensorflow查了一下cuDNN需要7.4版本而我安装的cuda 10.0对应的cuDNN是7.3 https://www.tensorflow.org/install/source#tested_build_configurations
不想再重新花费时间去安装cuda又搜了一下找到一个比较简单的解决方法就是使用conda安装tensorflowconda会自动配置对应版本号的cuda和cudnn
conda install tensorflow-gpu1.13.1查看安装结果cudnn版本7.6至此模型终于正常跑起来了
