Anaconda所有版本
https://repo.anaconda.com/
自己随意安装,minianaconda-4.8.2 python37
下载依赖VC++ 2015
https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads
查看显卡:命令行: nvidia-smi
安装对应的cudatoolkit版本
conda install cudatoolkit=11.0
安装tensorflow
需要安装的软件:https://www.tensorflow.org/install/gpu
需要查看安装版本 :https://tensorflow.google.cn/install/source#gpu
conda install cudatoolkit=11.0
pip install tensorflow-gpu==2.4.0 -i https://pypi.douban.com/simple/
pip install matplotlib notebook -i https://pypi.douban.com/simple/
测试是否完成
import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())
总结
版本一样即可
先安装CUDA,然后安装cuDNN,在安装VC++2015.
conda install cudatoolkit=11.0
pip install tensorflow-gpu==2.4.0 -i https://pypi.douban.com/simple/