人工智能-Windows安装TensorFlow

人工智能-Windows安装TensorFlow

安装配置Miniconda

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

安装配置pycharm

https://www.jetbrains.com/pycharm/download/

安装CUDA

//首先运行命令
conda install tensorflow-gpu
//先不要安装,查看TensorFlow支持的cuda版本,如:cudatoolkit-10.1.243
//然后按需求下载cuda
https://developer.nvidia.com/cuda-toolkit-archive
//最后按需求安装cuda,比如VS,document等就不需要了
//安装完成后运行命令验证安装结果
nvcc -V

安装CUDNN

//首先运行命令
conda install tensorflow-gpu
//先不要安装,查看TensorFlow支持的cudnn版本,如:cudnn-7.6.5
https://developer.nvidia.com/rdp/cudnn-archive
//下载完毕后,将所有文件拷贝进cuda根目录下
//这个不用验证

安装TensorFlow

//这玩意对python语言版本有要求
//运行命令查看要求,如果不符合,更换语言版本
conda search tensorflow-gpu
//运行命令直接安装
conda install tensorflow-gpu
//安装完成后运行命令验证安装结果
import tensorflow as tf
tf.test.gpu_device_name()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章