Anaconda中pytorch环境搭建(包括详细的虚拟环境创建,以及虚拟环境中jupyter notebook的使用)
?
PyTorch是一個開源的Python機器學習庫,基于Torch,用于自然語言處理等應用程序。
2017年1月,由Facebook人工智能研究院(FAIR)基于Torch推出了PyTorch。它是一個基于Python的可續計算包,提供兩個高級功能:1、具有強大的GPU加速的張量計算(如NumPy)。2、包含自動求導系統的深度神經網絡。
本次想使用Anaconda建立Pytorch虛擬環境來安裝Pytorch。(純屬個人強迫癥,不想讓所有的東西都放在最初的base環境下,所以想基于不同的框架來建立Anaconda虛擬環境)
為什么要使用Anaconda虛擬環境安裝Pytorch?
- 因為環境中通常需要安裝很多軟件,例如:我同時在使用tensorflow框架。但是他們所需要的Python的關聯模塊或版本會有所差異。如果都裝在一個環境中難免會引起沖突。所以,選擇虛擬環境能很好地避免環境之間的沖突。
1. 創建虛擬環境(名字pytorch)
1.1打開anaconda prompt
1.2我首先在C:\Users\TIMI這個目錄下創建了一個文件夾,叫做envs,
因為我暫時不確定在該路徑下安裝的虛擬環境,會存在哪里,所以提前建立這個文件路徑便于后期對比查看。
1.3之后執行命令
cd envs
如下:
1.4在此路徑下執行命令:
conda create --name pytorch python=3.7 anaconda
(PS:最后加anaconda是為了在這個虛擬環境(虛擬環境名字就叫pytorch)中把numpy, jupyter, pandas等一起安裝了,所以耗時較長,也可以不加)
之后就是等待安裝............
安裝結束的界面是如下:
此時查看之前建立的envs文件即C:\Users\TIMI\envs該目錄下下邊是空的。
而F:\softinstall\Study_Software\Anaconda\envs這個是Anaconda的安裝目錄下的envs保存的才是你創建虛擬環境的地方。
所以:一般如果我們不指定路徑的話,虛擬環境創建后會默認安裝在anaconda所安裝的目錄下,
1.5、創建好后,按照提示激活環境,輸入命令
conda activate pytorch
1.6、在虛擬環境中安裝ipykernel (切記這一步是在對應的虛擬環境中進行)
執行命令:
pip install ipykernel
1.7、使用ipykernel生成虛擬環境的kernel(這一步也是在虛擬環境中進行)
執行命令:
python -m ipykernel install --user --name 環境名
#即此處在自己的虛擬環境下是:
python -m ipykernel install --user --name pytorch
結果如下:
1.8、配置完成,退出虛擬環境在base環境中打開jupyter notebook,會發現已經有虛擬環境的kernel可供選擇。
執行命令:
conda deactivate
之后就是退出來了虛擬環境,回到了base環境下。
在base環境下輸入命令:jupyter notebook打開jupyter notebook。
此時查看你的右下角程序安裝標志下會發現:
選擇jupyter notebook(Anaconda)和jupyter notebook(pytorch)分別打開如下:
? ? ? ? ? ? ? ? ? ? ? ?分別可以讓你選擇要是用的環境。
或者kernel選擇自己的版本就可以了
查看兩個python的環境不同,也就是使用的Jupyter是在哪個環境下:
解釋:
?sys.executable返回python解釋器的路徑
sys.path是python的搜索模塊的路徑集,是一個list。?這兩個是在Anaconda下直接打開的juyter下執行的結果。
之后就是在這個環境下安裝自己需要的各種包了。(目前只是創建了一個虛擬環境而已)
2、安裝pytorch
首先在Anaconda prompt下輸入命令查看安裝的包
pip list
經查看沒有安裝pytorch,所以先安裝。
首先查看自己的電腦是NVIDIA的顯卡,再查看對應驅動版本以及CUDA的版本。
NVIDIA SMI是顯卡是一個硬件,所以需要驅動。Driver Version是驅動版本,CUDA Version是CUDA的版本。
CUDA是NVIDIA推出的只能用于自家GPU的并行計算框架。只有安裝這個框架才能夠進行復雜的并行計算。主流的深度學習框架也都是基于CUDA進行GPU并行加速的,幾乎無一例外。還有一個叫做cudnn,是針對深度卷積神經網絡的加速庫。
參考鏈接:https://blog.csdn.net/xiaoxiaolibai/article/details/104770046
下面是使用使用anaconda安裝pytorch----------------------------------------------》
Pytorch官方網站:https://pytorch.org/
進入:https://pytorch.org/get-started/locally/
選擇相應的你自己對應的版本號,會有命令出現
也可以采用其他方法安裝:Pip、LibTorch、Source。(見左圖自行選擇)
復制上述命令到你的Anaconda Prompt(pytorch)下,就可以執行安裝了:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
彈出提示,輸入 y,即可完成安裝,顯示“done”。
PS:看好執行命令的環境,我是在自己新創建的虛擬環境下。
執行之后就是等待安裝的過程。
安裝過程中遇到的問題:
安裝完成后,測試一下是否安裝成功:
import torch
import torchvision
print(torch.__version__)
如果以上步驟都沒問題,那么安裝成功。
3、安裝torchvision
上述過程已經附帶安裝過了。
總結
以上是生活随笔為你收集整理的Anaconda中pytorch环境搭建(包括详细的虚拟环境创建,以及虚拟环境中jupyter notebook的使用)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Anaconda使用
- 下一篇: 卷积神经网络CNN---权值共享