tensorflow 1.15.0编译时如何添加对GPU的支持
收藏回复举报
tensorflow 1.15.0编译时如何添加对GPU的支持
t('forum.solved') 已解决
发表于2024-03-29 17:56:32
0 查看

环境:已安装驱动、固件、cann-toolkit、cann-tfplugin,300I pro推理卡

编译tensorflow 1.15.0版本时,按照如下链接编译安装:

https://www.hiascend.com/document/detail/zh/quick-installation/23.0.0/quickinstg_train/900_A2_PoDc/quickinstg_900A2_PoDc__0058.html#ZH-CN_TOPIC_0000001735835956

开始未添加ROCm和cuda支持,编译成功后跑图片分类任务,发现使用的是CPU而不是GPU,tf.test.is_gpu_available()返回false

尝试重新编译,添加cuda支持环境没有nvidia卡找不到文件,请问如何添加对Ascend NPU卡的支持?

我要发帖子