anaconda建立虛擬環境安裝不同深度學習框架

2021-10-01 18:00:25 字數 2876 閱讀 6062

最近帶乙個學弟做畢設,深度學習方向,需要安裝tensorflow、pytorch等深度學習框架。想當初自己是新手的時候可沒少被配環境、安裝深度學習框架折騰…吃一塹長一智,趁著帶畢設,重新回顧一下配環境、裝框架那些事兒,並形成文件,方便後期檢視~~

使用anaconda安裝tensorflow、caffe、pytorch等深度學習框架很方便,大致思路為:首先通過anaconda建立虛擬環境(如果沒有安裝過anaconda,先要安裝anaconda);其次在不同的虛擬環境下安裝不同版本的深度學習框架(例如在乙個虛擬環境上安裝乙個tensorflow1.4,在另乙個虛擬環境中上安裝乙個tensorflow1.6,在另乙個虛擬環境中安裝pytorch),這樣做的好處是一旦配置出錯,可以直接刪除虛擬環境包,而不影響整個系統。

在系統上安裝anaconda

通過anaconda建立虛擬環境

conda create -n py3tf pip python=3.5
上述**建立了乙個名叫py3tf的虛擬環境(環境名字可以隨便起,但最好包含tf、caffe或pytorch和對相應的版本字樣,以免日後混淆),並在這個環境裡安裝了python3.5。

【注:】這裡的python3.5和anaconda的版本沒有對應關係,假設通過anaconda安裝的python版本是3.6,但是這裡也可以安裝python3.5。

通過如下命令進入虛擬環境

source activate py3tf
這就進入了虛擬環境,進入後你會發現命令列和之前的有所不同,這時命令列前面多了括號,括號裡面含有虛擬環境的名字,之後在這種狀態下安裝的所有內容就都會安裝在虛擬環境裡,退出虛擬環境後,將不能使用該虛擬環境中安裝的東西。

[global]

index-url =

**conda源設定:**在終端輸入

conda config --add channels 

conda config --set show_channel_urls yes

如果你的系統中已經安裝了cuda和cudnn,這一步就不需要執行了,虛擬環境需要使用cuda和cudnn時會自動呼叫系統中的。但是如果你的系統的cudnn和cuda版本和所要安裝的tensorflow或者pytorch不匹配,就需要在虛擬環境中重新安裝,安裝方法如下:

conda install cudnn=7.1.2
通過上述命令安裝cudnn,會自動安裝依賴項 cuda(也可以自己通過conda install cuda =9.0自己安裝),這裡我安裝的cudnn版本是7.1.2。如果安裝不成功的話,可以嘗試加入清華的源(上述命令 加入 -c 源位址)。

安裝tensorflow

pip install --ignore-installed --upgrade
後面給的位址是gpu版的tensorflow(linux環境,gpu版本,cp35-cp35m表示python的版本是python35,tensorflow的版本是1.12.0)。這裡是軟體包位置找到與python版本一致的url放到這裡。

【注:】有時候上述命令一次不行,需要再試一次,再不行的話就要考慮cuda與cudnn的問題了。

安裝keras

pip install keras
安裝opencv

pip install opencv-python
測試安裝的內容

在虛擬環境下輸入python,進入python環境,看看包是否可以匯入

import tensorflow as tf  # 檢視是否可以匯入tensorflow

tf.__version__ # 檢視tensorflow版本

import keras # 檢視是否可以匯入keras

keras.__version__ # 檢視keras版本

import cv2 # 檢視是否可以匯入opencv

cv2.__version__ # 檢視opencv版本

安裝pytorch

建立虛擬環境和配置cuda與cudnn跟上述是一樣的,使用source activate name(自己定義的虛擬環境名字)進入虛擬環境就以後,需要去pytorch官網按照你的系統環境選擇相應的作業系統(win or linux)、conda版本、python版本、cuda版本,響應版本選擇好後會給出安裝命令,如下:

conda install pytorch torchvision -c pytorch
測試pytorch

import torch

torch__version__

檢視已安裝的虛擬環境

conda env list
進入虛擬環境

source activate 虛擬環境名稱
退出虛擬環境

source deactivate
刪除虛擬環境

conda remove -n 虛擬環境名稱 --all
刪除虛擬環境中的包

conda remove --name $虛擬環境名稱  $包名稱
檢視包

conda list
創造乙個相同環境

conda create --name py3tf16gpu --clone py3tf   #根據py3tf16創造乙個名稱為py3tf16gpu的虛擬環境

Anaconda建立虛擬環境

1.虛擬環境在anaconda資料夾的envs資料夾裡 建立乙個名為env,python版本為2.7的虛擬環境 conda create n env python 2.7 conda create name env python 2.7啟用 退出虛擬環境 conda activate env con...

Anaconda建立虛擬環境

python 虛擬環境用於將軟體包安裝與系統隔離開來。conda 建立乙個新的虛擬環境,方法是選擇 python 直譯器並建立乙個 venv 目錄來存放它 conda create n venv pip python 3.6 select python version啟用虛擬環境 source ac...

Anaconda 建立虛擬環境

anaconda是乙個用於科學計算的python發行版,提供了包管理與環境管理的功能,可以很方便地解決多版本python並存 切換以及各種第三方包安裝問題。1.anaconda與conda區別 conda可以理解為乙個工具,也是乙個可執行命令,其核心功能是包管理與環境管理。包管理與pip的使用類似,...