996Worker
996Worker
发布于 2023-01-03 / 117 阅读
0
0

Windows配置Tensorflow的GPU运行环境

Details

草泥马CSDN等等垃圾网误导人,说什么配置GPU环境要手动安装各种CUDA,cudnn之类的, 一旦版本配对不上就直接寄。

傻逼。

其实,搞出来GPU运行环境,只需要几步。

Steps

  1. 安装 Conda, Anaconda;
  2. 去找需要的tensorflow-gpu版本:https://anaconda.org/search?q=tensorflow-gpu;
  3. 打开anaconda prompt;
  4. 以安装tensorflow 2.6于conda环境tf为例,在anaconda prompt输入"conda create -n tf tensorflow-gpu=2.6.0";
  5. 等待安装完成;
  6. 测试GPU是否可用:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

结论

配置GPU环境其实很简单,去你大爷的CSDN和灌水博客。


评论