Conda配置CUDA、cudnn環境,Tensorflow、CUDA、cudnn與Tensorflow、keras版本對應關係

Conda配置CUDA、cudnn環境, Tensorflow、CUDA、cudnn與Tensorflow、Keras版本對應關係

Conda配置CUDA、cudnn環境

使用conda創建python虛擬環境,可以每個虛擬環境獨立一個CUDA版本,不使用虛擬環境如果算法環境要求的CUDA版本不一樣,就需要重新安裝CUDA,所以做深度學習還是推薦使用conda

# CUDA
# conda install cudatoolkit=版本號
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
# cudnn
# conda install cudnn=版本號
conda install cudnn=7.3.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

Tensorflow、CUDA、cudnn版本對應關係

Linux:
linux
Windows:
windows

Tensorflow與Keras版本對應關係

tf與keras

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章