我的tensorflow-gpu 安装成功了吗?
回答列表
-
2020/05/24 下午 08:00骑士贊同數:0不贊同數:0留言數:0
不好意思,我补充一下我的问题。
我按照网络上的教程安装了多个版本的tensorflow-gpu和tensorflow的虚拟环境,导致我不清楚是否已经成功使用上tensorflow-gpu。请帮忙按照我正文中的截图内容来判定一下是否有问题?真的弄了好久,如果有问题能否提供新的参考资料让我重新安装一下?非常感谢
-
2020/05/25 下午 00:41Jeffrey贊同數:0不贊同數:0留言數:0
Hello, 從訊息來看, CUDA 安裝沒有成功, Tensorflow GPU 有安裝但是抓不到GPU.
有關 CUDA, 有些步驟:
1. CUDA 安裝
要確認一下, 顯卡能支持
2. cuDNN
請根據你的CUDA版本選擇對應版本下載!
解壓縮後,將三個資料夾中的檔案,個別複製到CUDA的相對應資料夾中。
cuDNN下載:https://developer.nvidia.com/cuda-toolkit-archive
3. 微軟的 Vitual studio
安裝 tensorflow-gpu
https://tensorflow.google.cn/install/gpu?hl=zh-cn
必須在系統中安裝以下 NVIDIA® 軟體:
- NVIDIA® GPU 驅動程式:CUDA 10.1 需要 418.x 或更高版本。
- CUDA® 工具套件:TensorFlow 支援 CUDA 10.1(TensorFlow 2.1.0 及更高版本)
- CUDA 工具套件附帶的CUPTI。
- cuDNN SDK(7.6 及更高版本)
- (可選)TensorRT 6.0,可縮短在某些模型上進行推斷的延遲並提高輸送量。
-
2020/05/26 下午 09:49骑士贊同數:0不贊同數:0留言數:0
谢谢您的整理回答:不过我已经按照好了NVIDIA® GPU 驅動程式,CUDA® 工具套件以及cuDNN SDK。如下截图所示。另外您提供的Tensorflow GPU的安装是限Linux系统的(我的是window系统)。请问还有其它解决方法吗,谢谢?
-
2020/05/26 下午 10:16骑士贊同數:0不贊同數:0留言數:0
刚刚又用各种方式测试了一些结果,如下,请问老师这是什么原因?
显示没有使用GPU的结果如下:
显示有使用GPU的结果如下
-
2020/05/27 下午 05:12Jeffrey贊同數:0不贊同數:0留言數:3
建議把 GPU device mapping 到 cell 裡面, 參考如下:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# os.getcwd()
import tensorflow as tf
W = tf.constant([1.0, 2.0, 3.0, 4.0], shape=[2, 2], name='W')
x = tf.constant([1.3, 2.4], shape=[2, 1], name='x')
y = tf.matmul(W, x)
# 只使用 30% 的 GPU 記憶體
gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction = 0.3)
# Creates a session with log_device_placement set to True.
sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))
PS: 若有問題, 請把 tensorflow 移除, 只保留tensorflow-gpu