win10 安装tensorflow(GPU)

最新推荐文章于 2024-06-06 00:48:03 发布

原创最新推荐文章于 2024-06-06 00:48:03 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

在Windows 10上安装TensorFlow的GPU版本，需配合Python3、CUDA8.0和CuDNN6.0。安装完成后，通过运行Python代码验证GPU成功启用，显示设备为GeForce MX150，内存和计算能力详细信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://developer.nvidia.com/cuda-downloads

安装很简单，用python3，cuda8.0，cudnn6.0 即可。

import tensorflow as tf
>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
2018-01-16 13:05:20.553627: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2018-01-16 13:05:22.892739: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1030] Found device 0 with properties:
name: GeForce MX150 major: 6 minor: 1 memoryClockRate(GHz): 1.341
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.63GiB
2018-01-16 13:05:22.892884: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce MX150, pci bus id: 0000:01:00.0, compute capability: 6.1)
Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce MX150, pci bus id: 0000:01:00.0, compute capability: 6.1
2018-01-16 13:05:22.965794: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\direct_session.cc:299] Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce MX150, pci bus id: 0000:01:00.0, compute capability: 6.1

可以看到GPU的信息了。