win10安裝tensorflow-gpu版本遇到的坑
阿新 • • 發佈:2019-02-20
ImportError: Could not find 'cudart64_90.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit
1,版本過高,需要cuda9.0。但是我電腦是安裝的cuda8.0。於是我準備安裝低版本,利用清華的資源,安裝1.3版本,利用下面的命令。
pip install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl
但是又出現下面的報錯。
tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl is not a supported wheel on this platform.
原因是我用的python3.6,這個版本是3.5的,所以又換了一個版本。用下面的命令。
pip install --upgrade https://mirrors.tuna .tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.2.1-cp36-cp36m-win_amd64.whl