【超详细】pytorch安装及配置
文章目錄
- 前言
- 一、Anaconda安裝
- 二、PyTorch的下載及配置
- 1.創建虛擬環境
- 2.在虛擬環境中下載PyTorch
- 3.本機CUDA版本低怎么下載CUDA高版本的pytorch
- 4.Jupyter Notebook調用PyTorch虛擬環境
- 5.PyCharm調用PyTorch虛擬環境
- 6. Jupyter Notebook使用中碰到的問題
- 6.1輸入jupyter notebook后顯示無法定位程序輸入點
- 6.2創建好的虛擬環境在Jupyter Notebook中使用不了
- 總結
前言
本周準備做一個基于PyTorch深度學習實踐的專欄,咱們需要先將環境配置好,所以本文先介紹一下如何下載和配置PyTorch
一、Anaconda安裝
Anaconda的下載安裝比較簡單,網上的教程都是保姆級的,大家可以看看這個博主寫的這篇博客Anaconda下載安裝介紹,里面還解決了Anaconda Navigator打不開的問題,親測有效哦!
二、PyTorch的下載及配置
1.創建虛擬環境
在下載PyTorch之前,首先我們需要創建一個虛擬環境,在電腦左下角的“開始”欄中找到Anaconda Prompt,點擊打開
輸入以下代碼conda create -n pytorch2 python=3.7
代碼中的pytorch2是我給這個虛擬環境取的名字,大家可以自己定義取的名字,后面的python=3.7是我所下載的python版本,這個需要根據大家下載的python版本寫。
輸入代碼回車后,會出現([y]/[n]?),輸入y等待安裝完畢后,我們的虛擬環境便創建好了。
2.在虛擬環境中下載PyTorch
創建完虛擬環境后,打開Anaconda Prompt,輸入activate pytorch2,激活所創建的虛擬環境。
然后進入到pytorch的官網:PyTorch官網
電腦沒有顯卡的選擇cpu版本,有顯卡的可以選擇CUDA版本的(cpu版本的在后續大量數據訓練中速度肯定會更慢一點)
打開cmd控制臺,輸入nvidia-smi查看本機的cuda版本
我這里是CUDA11.7的,可以向下兼容,所以在pytorch官網我下載CUDA11.3的pytorch
復制這行命令,在剛剛創建的虛擬環境中輸入
會出現([y]/[n]?),輸入y,然后等待它下載完成即可。
下載完成后,我們驗證一下是否成功下載好了PyTorch。
同樣的,在這個虛擬環境中輸入python進入python環境,然后輸入import torch如果沒有報錯則下載成功,輸入print(torch.__version__)可以查看pytorch版本,最后檢查一下cuda是否可以用于當前版本的pytorch,輸入torch.cuda.is_available(),若為True,則表示可用。最后輸入exit()退出python環境。
3.本機CUDA版本低怎么下載CUDA高版本的pytorch
我的電腦最開始的cuda版是11.1,而不是上面顯示的11.7,這是怎么一回事呢?原來電腦所支持的cuda版本是可以升級的,我的顯卡是英偉達的,打開NVIDIA控制面板查看顯卡類型,我的是GeForce GTX 1650 Ti
然后登陸英偉達的官網:NVIDIA官網
選擇與自己電腦顯卡型號相同的選項(帶Notebooks的是筆記本),然后點擊搜索
最后點擊下載即可。后面的安裝教程也比較簡單,這里就不再講訴了,大家可以看看這篇博客。https://blog.csdn.net/m0_37870649/article/details/105356540
安裝完成后,再次查看電腦的cuda版本可以發現已經升級到最新,此時可以按照之前的方法下載CUDA為11.3的pytorch。
4.Jupyter Notebook調用PyTorch虛擬環境
Jupyter Notebook可以簡單的將其理解為一個編程開發工具,就是你寫代碼的地方。成功安裝anaconda和pytorch后,我們如何在Jupyter Notebook中使用?
大家要明白的是Jupyter Notebook本身是默認使用一種anaconda中root目錄下的Python環境的,如果想使用其它的虛擬環境,還需要通過插件來實現,也就是nb_conda插件。
首先,打開anaconda prompt 輸入conda install nb_conda安裝該插件
然后進入想要添加到jupyter notebook中的虛擬環境,比如我這里是進如剛剛創建的pytorch2虛擬環境,即輸入activate pytorch2,之后在該虛擬環境下輸入conda install ipykernel(這行代碼得在虛擬環境中輸入),等待安裝就行了。
在虛擬環境中輸入jupyter notebook就會跳轉到其界面
5.PyCharm調用PyTorch虛擬環境
PyCharm和Jupyter Notebook的功能是一樣的,如果你覺得jupyter notebook不好使的話,也可以選擇在PyCharm中寫代碼,下面是在PyCharm中調用Pytorch虛擬環境
只需要在Settings中將Python Interpreter 換成你剛剛創建的虛擬環境即可,具體步驟如下:
點擊小齒輪中的Add,然后選擇Conda Environment
以上就是在PyCharm中調用PyTorch
6. Jupyter Notebook使用中碰到的問題
6.1輸入jupyter notebook后顯示無法定位程序輸入點
解決方法:直接將這個報錯下的文件刪除就不會再報錯了,并不影響Jupyter Notebook的使用
6.2創建好的虛擬環境在Jupyter Notebook中使用不了
如下圖,虛擬環境pytorch2上帶了個星號,點進去是無法正常運行程序的
如下圖,該圓圈的意思是內核正忙,無法使用
問題產生原因可能是文件路徑包含中文或者用戶名中包含中文
(如下圖所示,當用戶名中包含中文,仍然會導致 jupyter notebook無法運行。進入jupyter notebook之前只會彈出以下控制臺窗口,并不會自動打開jupyter notebook操作窗口)
這個博主的博客里寫了解決方法:
就是新建一個不帶中文的用戶,登錄新的賬戶,在新的賬戶下安裝jupyter notebook或者使用虛擬機,具體步驟可以參考這個博主的文章解決Jupyter Notebook內核正忙的方案
總結
本文主要介紹了pytorch的下載安裝和配置,以及遇到的一些問題和解決方法。重點介紹了如何下載CUDA版的pytorch,以及升級本機CUDA的方法,并且講解了如何在Jupyter Notebook和PyCharm中使用我們所創建的虛擬環境。
總結
以上是生活随笔為你收集整理的【超详细】pytorch安装及配置的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: pytorch历史版本下载
- 下一篇: 5W无线充非认证版便宜实用无线充电器常用