В хранилище есть новый файл колеса:
https://github.com/fo40225/tensorflow-windows-wheel
Следующий файл работает очень хорошо:
https://github.com/fo40225/tensorflow-windows-wheel/blob/master/2.0.0/py37/GPU/cuda101cudnn76sse2/tensorflow_gpu-2.0.0-cp37-cp37m-win_amd64.whl
Как указано в файле Readme.md:
"Для компиляции при первом запуске TensorFlow потребуется время."
Посмотрите на этот тест:
>>>import tensorflow as tf
tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
>>>print(tf.__version__)
2.0.0
>>>from tensorflow.python.client import device_lib
>>>print(device_lib.list_local_devices())
tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties:
name: GeForce GTX 1050 major: 6 minor: 1 memoryClockRate(GHz): 1.531
GPU libraries are statically linked, skip dlopen check.
Adding visible gpu devices: 0
Device interconnect StreamExecutor with strength 1 edge matrix:
0
0: N
Created TensorFlow device (/device:GPU:0 with 1340 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 4456898788177247918
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 1406107238
locality {
bus_id: 1
links {
}
}
incarnation: 3224787151756357043
physical_device_desc: "device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1"
]