Windows下不使用Anaconda安装Tensorflow-gpu简述

  1. 安装CUDA Tookit,进入自定义,在电脑已有显卡驱动的情况下,只安装第一大项CUDA(其中的VS和Nsight相关内容也不用装)
  2. 下载对应版本的cuDNN,将其解压合并到CUDA安装文件夹中
  3. 安装对应版本的Python和Tensorflow,测试
import tensorflow as tf
gpus = tf.config.list_physical_devices("GPU")
print(gpus)
posted @ 2021-12-24 16:02  GeraintJht  阅读(158)  评论(0编辑  收藏  举报