首先是3者以及python的版本关系,如下图

在这里插入图片描述

如果是用anaconda搭建环境建议在conda的命令行界面(Anaconda Prompt)下载这3个(楼主去官网手动下载CUDA和CUDnn,结果和conda里的环境一直连接不上,各种重装搬运都是血泪史啊)

前面两句话是搭房子应该不用说了,后面三句话就是安装cuda、cudnn、tf-gpu最好按顺序来,可能我最开始几次安不上去也有先安装tensorflow-gpu后安装cuda的原因

conda create -n py36 python=3.6

conda activate py36


conda install cudatoolkit=11.0

conda install -c conda-forge cudnn=8.0

pip install tensorflow-gpu=2.4.0

参考:conda环境下安装tensorflow-gpu 2.4 + opencv 3.3_西界的博客-CSDN博客

如果不是anaconda,用pip的话,这篇文章个人感觉不错。而且他下面给了一个测试cuda是否可用,GPU是否被调用的代码(放到pycharm就行,不分conda还是别的都可以试试),真滴良心

import tensorflow as tf

print(tf.__version__)  # 查看tensorflow版本
print(tf.__path__)     # 查看tensorflow安装路径

a = tf.test.is_built_with_cuda()  # 判断CUDA是否可以用
b = tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)  # 判断GPU是否可以用

print(a) # 显示True表示CUDA可用
print(b) # 显示True表示GPU可用

# 查看驱动名称
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

————————————————
版权声明:本文为CSDN博主「落叶阳光」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/xiangxiang613/article/details/112603083

参考:为什么CUDA装好了,Tensorflow-GPU不能用?_xiangxiang613的专栏-CSDN博客_安装了cuda却不调用gpu

还有一些GPU的问题

一个是有多个GPU的时候如何指定GPU,一个简单的办法就是用NVIDIA的控制面板指定默认GPU

 这样就好啦!

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐