遇到的問題:
代碼要求CUDA9.0+tensorflow1.4.0,但是用的伺服器沒有權限,不能改任何東西,被迫無奈,選擇了合适的版本解決沖突。
一、這個問題是你的cuda跟tensorflow版本沖突問題
(nvcc -V 檢視cuda版本
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
)可以看一下對應版本
二、我用的伺服器cuda10.0.130,沒有權限等等問題,隻能改tensorflow(建議建一個自己的環境)
直接複制
pip install tensorflow-gpu==1.14(版本随便改,對應上面的)
三、運作代碼完美解決。