人工智能-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()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章