Tensorflow 在WSL2上运行Pycharm:无法访问GPU

我终于能够让Pycharm在WSL2上工作了。然而，我无法从tensorflow内部调用“GPU”

我尝试调用此代码段： 将tensorflow.compat.v1导入为tf#以获得tf2.0中类似于tf1.x的行为 tf.disable_v2_behavior（）

因为我需要GPU，所以代码没有运行。但是，如果我将“/gpu:0”更改为“/cpu:0”，它将运行

2我安装了tensorflow gpu

此外，WSL2上的Docker成功地执行了GPU的测试。因此，cuda库已正确安装。结果部分如下：

GPU Device 0: "Quadro RTX 4000" with compute capability 7.5
Compute 7.5 CUDA device: [Quadro RTX 4000]
40960 bodies, total time for 10 iterations: 81.261 ms
= 206.462 billion interactions per second
= 4129.240 single-precision GFLOP/s at 20 flops per interaction

任何帮助都将不胜感激

GPU Device 0: "Quadro RTX 4000" with compute capability 7.5
Compute 7.5 CUDA device: [Quadro RTX 4000]
40960 bodies, total time for 10 iterations: 81.261 ms
= 206.462 billion interactions per second
= 4129.240 single-precision GFLOP/s at 20 flops per interaction