- 安装CUDA Tookit,进入自定义,在电脑已有显卡驱动的情况下,只安装第一大项CUDA(其中的VS和Nsight相关内容也不用装)
- 下载对应版本的cuDNN,将其解压合并到CUDA安装文件夹中
- 安装对应版本的Python和Tensorflow,测试
import tensorflow as tf
gpus = tf.config.list_physical_devices("GPU")
print(gpus)
import tensorflow as tf
gpus = tf.config.list_physical_devices("GPU")
print(gpus)