参考官网地址:
传送门
如需支持 GPU,请将 CUDA 和 cuDNN bin 目录添加到 $PATH 中:
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.0/bin:$PATH"
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.0/extras/CUPTI/libx64:$PATH"
export PATH="/c/tools/cuda/bin:$PATH"
进入python编译环境,输入一下代码,如果结果是True,表示GPU可用
import tensorflow as tf
print(tf.test.is_gpu_available())