Chr*_*hmé 1 python tensorflow tensorflow2.0
我有两台配备 NVIDIA GeForce RTX 3060 GPU 的 Windows 11 笔记本电脑,我想在其上运行 Tensorflow。
如果这很重要的话,两台笔记本电脑都是 Lenovo Legion 5 笔记本电脑,“GPU 工作模式”设置为“混合自动模式”。
第一台笔记本电脑具有以下设置:
Python 3.10.7
Tensorflow 2.9.1
CUDA 11.2.0
cuDNN 8.1.1
CPU AMD Ryzen 7 6800H
GPU0 NVIDIA GeForce RTX 3060
GPU1 AMD Radeon Graphics
Run Code Online (Sandbox Code Playgroud)
第二台笔记本电脑具有以下设置:
Python 3.10.9 Virtual Environment
Tensorflow 2.11.0
CUDA 11.2.2
cuDNN 8.1.1
CPU Intel Core i7 12th Gen 12700H
GPU0 Intel Iris Xe
GPU1 NVIDIA GeForce RTX 3060
Run Code Online (Sandbox Code Playgroud)
CUDA 和 cuDNN 是按照此视频安装的:https://www.youtube.com/watch?v =hHWkvEcDBO0(conda 部分除外)。
在第一台笔记本电脑上,一切正常。但在第二次执行时tf.config.list_physical_devices('GPU'),我得到一个空列表。
我尝试将CUDA_VISIBLE_DEVICES变量设置为"0"某些人在其他帖子中提到的那样,但它不起作用。
我还在第三台笔记本电脑上尝试了与第二台笔记本电脑相同的操作,但遇到了同样的问题。
可能是什么问题呢?
实际上,问题在于您使用的是 Windows,TensorFlow 2.11 和更新版本不再对 Windows 上的 GPU 提供本机支持,请参见 TensorFlow网站:
注意:TensorFlow 2.10 是在本机 Windows 上支持 GPU 的最后一个 TensorFlow 版本。从 TensorFlow 2.11 开始,您需要在 WSL2 中安装 TensorFlow,或者安装 tensorflow-cpu,并且可以选择尝试 TensorFlow-DirectML-Plugin
从 TensorFlow 2.11 开始,在 Windows 上获得 GPU 支持的唯一方法是使用WSL2。
| 归档时间: |
|
| 查看次数: |
7905 次 |
| 最近记录: |