1. 程式人生 > >win10安裝tensorflow-gpu版本遇到的坑

win10安裝tensorflow-gpu版本遇到的坑

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