环境:已安装驱动、固件、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卡的支持?
环境:已安装驱动、固件、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卡的支持?