首先是安装anaconda,这个是最基本的,官网下载:link
安装过程:无坑,随便搜索一个安装过程便可以,记住把环境变量的对号勾上就行了。
tensorflow:注意好GPU版本还是CPU版本
注意其与cuda以及python的对应版本,自行搜索对应关系,中文官网;link,列举下我当下最新的对应关系。具体安装教程自行搜索,本文不做细致的讲解。
CUDA安装:本人建议先安装VS2019,这可以为小白省去大量的时间,大牛无视
安装对应的CUDA,此次建议自行搜索对应的显卡对应版本
安装cudnn时需要注意选好对应的系统,同时看后缀别选错了其余CUDA的版本对应
测试代码
import tensorflow as tf
import os
os.environ['TF_XLA_FLAGS'] = '--tf_xla_enable_xla_devices'
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
print(tf.__version__)
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())
看是否显示正确,在这些成功中有一个显示失败的便复制代码搜索,基本都能解决