我正在尝试使用以下设置在 Windows 10 上运行 TensorFlow:
Anaconda3 与
蟒蛇3.8
张量流2.2.0
显卡:RTX3090
cuda_10.1.243
cudnn-v7.6.5.32 适用于 windows10-x64
运行以下代码需要 5 到 10 分钟才能打印输出。
import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
Run Code Online (Sandbox Code Playgroud)
我立即得到以下输出,但在继续之前它会挂起几分钟。
1-17 04:03:00.039069: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-11-17 04:03:00.042677: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2020-11-17 04:03:00.045041: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2020-11-17 04:03:00.045775: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2020-11-17 04:03:00.049246: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2020-11-17 04:03:00.050633: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened …Run Code Online (Sandbox Code Playgroud)