在 VSCode 中使用 GPU 进行 tensorflow答案

【问题标题】：Use GPU in VSCode for tensorflow在 VSCode 中使用 GPU 进行 tensorflow
【发布时间】：2022-12-16 11:39:27
【问题描述】：

我关注了这个stackoverflow answer。然后，我尝试将此 nvidia steps (conda install cuda -c nvidia) 安装在 conda 环境中。但是当我尝试

import tensorflow as tf
tf.config.list_physical_devices('GPU')
# output: []

我有一个 3070 GPU。我究竟做错了什么？

【问题讨论】：

按照这些步骤使用 conda 安装 Tensorflow-gpuconda create -n tf_gpu python=3.9和conda activate tf_gpu以及conda install cudatoolkit==11.2和pip install tensorflow。验证安装 import tensorflow as tf 和 print(len(tf.config.list_physical_devices('GPU')))。谢谢

【解决方案1】：

访问https://www.tensorflow.org/install/pip并按照程序进行。您很可能没有提供某些软件要求。当我没有安装 cuda 工具包时，我遇到了同样的问题。安装 cuda 和 cudnn 后，它对我来说是固定的。

【讨论】：