Window&Linux多CUDA環境安裝不用愁啦

Windows 安裝 conda安裝 tf2.0

由於tf2須要cuda10,本人一直在用cuda9.2的1.x版本tensorflow。

此方法能有效避免系統python版本和pip包覆蓋問題,還能成功安裝多個版本的cuda

安裝工具:Miniconda、PC

1.下載conda(miniconda更簡潔,節省內存)

多個版本下載地址,自取~python

Windows 下載地址

https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-4.7.10-Windows-x86_64.exeshell

Linux 下載地址

https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-4.7.12-Linux-x86_64.sh工具

Mac 下載地址

https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-4.7.10-MacOSX-x86_64.pkg測試

2.安裝Miniconda

在這裏插入圖片描述

主要要注意這個地方,其餘地方默認設置就好:若是是自帶有python,請不要勾選,避免出現python版本覆蓋問題;第一個選項能夠不選,但後期要本身再設置

這裏路徑我選的是D:\Miniconda3

3.設置conda安裝源

安裝源文件位置:

Windows 源文件路徑url

C:\Users\Administrator.condarc

Linux 源文件路徑spa

~/.condarc
(1)打開 Anaconda Prompt
(2)清華源配置

依次在 Anaconda Prompt 或 Terminator中輸入命令行

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes
(3)本人安裝成功配置

直接複製便可食用哦~3d

ssl_verify: true
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
show_channel_urls: true

4.conda指令建立環境

四步走:建立—激活—安裝—測試
(1)建立新的conda環境

安裝python36虛擬環境,由於是gpu環境取名tf2gcode

conda create -n tf2g python=3.6
(2)激活環境
conda activate tf2g
(3)安裝cuda及tensorflow2.0

若存在舊版本cuda,直接到如下官網連接下載安裝不一樣版本的CUDA和cudnn就OK。安裝cpu版直接第二條指令不加gpublog

手動安裝CUDA和cudnn

下載cuda 10

cuda 10.0下載網址

https://developer.nvidia.com/cuda-10.0-download-archive

cudnn下載網址(找到相對應的cuda版本下載,需註冊)

https://developer.nvidia.com/rdp/cudnn-archive

精簡安裝便可,複製cudnn 7.6.5的三個文件夾到以下路徑
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
本人同時存在的多個CUDA版本

cuda 9.2 與 cuda10.0共存
在這裏插入圖片描述

這裏不用擔憂CUDA版本衝突,不一樣虛擬環境對應的tensorflow-gpu會找到對應的CUDA版本使用

這條指令經嘗試會報 CUDA版本不對應的錯誤,如圖

conda install cudatoolkit=10.0 cudnn
在這裏插入圖片描述

tensorflow2.0安裝

conda install tensorflow-gpu==2.0.0
(4)測試環境

當前命令行輸入python,進入python執行如下指令:

python
import tensorflow as tf    # tensorflow包導入
tf.__version__      # 查看版本
tf.test.is_gpu_available()  # 檢查是否能夠GPU加速

在這裏插入圖片描述

友情提示:

精簡安裝cuda時已自動配置好環境變量,只須要在conda中配置新的虛擬環境並下載安裝對應的CUDA版本便可。

分別顯示 2.0.0和True ,如圖安裝成功!

相關文章
相關標籤/搜索