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:
Windows: