關(guān)于Torch?torchvision?Python版本對(duì)應(yīng)關(guān)系說(shuō)明
1. torch- torchvision- python版本對(duì)應(yīng)關(guān)系
2. CUDA Toolkit 和PyTorch的對(duì)應(yīng)關(guān)系
3. 安裝說(shuō)明
3.1 用anaconda安裝pytorch
anaconda新建虛擬環(huán)境后，直接在pytorch官網(wǎng)官網(wǎng)鏈接找到“Install”按鈕。這里一鍵搞定torch，torchvision，cudatoolkit等等。
- 前提需要安裝好NVIDIA驅(qū)動(dòng)。
- 不需要另外安裝CUDA（筆者在沒有單獨(dú)安裝CUDA情況下，成功運(yùn)行了torch-gpu）。
- 和機(jī)器已經(jīng)裝好的CUDA不發(fā)生沖突。
- anaconda會(huì)自動(dòng)對(duì)應(yīng)torch，torchvision，cudatoolkit等的版本。
- 這里一般下載最新的torch版本。
原因是anaconda將torch，torchvision，cudatoolkit等等都集成在虛擬環(huán)境里，統(tǒng)一管理依賴包。
有鏡像源的情況下，去掉末尾的“-c pytorch”會(huì)更快，否則經(jīng)常容易下載超時(shí)。conda下載超時(shí)時(shí)長(zhǎng)可以設(shè)置。
conda config --show conda config --set remote_connect_timeout_secs 40 conda config --set remote_read_timeout_secs 100
conda config --show #查看conda設(shè)置 conda config --set #設(shè)置對(duì)應(yīng)的參數(shù)
3.2 不用Anaconda來(lái)管理環(huán)境
仔細(xì)對(duì)照torch、torchvision、cuda之間的對(duì)應(yīng)關(guān)系，還有NVIDIA和CUDA版本的關(guān)系。
3.3 對(duì)NVIDIA驅(qū)動(dòng)的要求
和NVIDIA驅(qū)動(dòng)直接關(guān)聯(lián)的是CUDA的版本。
如果安裝的是CUDA=10.0.130,那么在Windows系統(tǒng)要求NVIDIA驅(qū)動(dòng)大于411.31,不需要嚴(yán)格等于411.31。
如果小于411.31，則會(huì)報(bào)錯(cuò)顯示驅(qū)動(dòng)版本過舊。
筆者是將NVIDIA驅(qū)動(dòng)更新至最大456.38，正常使用。
3.4 下載 .whl 文件離線安裝
https://download.pytorch.org/whl/torch_stable.html
根據(jù)前面的對(duì)應(yīng)關(guān)系，下載好適合你的版本的 torch 、torchvision。
cu102 # 表示CUDA=10.2 cp37 # 表示python=3.7 linux or win?
下載好后，用pip安裝，先cd 到下載的文件夾
pip install torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl pip install torchvision-0.8.0-cp36-cp36m-win_amd64.whl
測(cè)試GPU版本的torch是否安裝成功
(torch) D:\MyData\xiaCN\Desktop\Work\unbiased> python Python 3.6.13 (default, Feb 19 2021, 05:17:09) [MSC v.1916 64 bit (AMD64)] on win32 Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> torch.cuda.is_available() True
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
在Python3.74+PyCharm2020.1 x64中安裝使用Kivy的詳細(xì)教程
這篇文章主要介紹了在Python3.74+PyCharm2020.1 x64中安裝使用Kivy的詳細(xì)教程，本文通過圖文實(shí)例相結(jié)合給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-08-08
python?pandas分割DataFrame中的字符串及元組的方法實(shí)現(xiàn)
本文主要介紹了python?pandas分割DataFrame中的字符串及元組的方法實(shí)現(xiàn)，主要介紹了3種方法，具有一定的參考價(jià)值，感興趣的可以了解一下2022-03-03
python+lunarcalendar庫(kù)實(shí)現(xiàn)使用農(nóng)歷日期
這篇文章主要為大家詳細(xì)介紹了python如何通過lunarcalendar庫(kù)實(shí)現(xiàn)使用農(nóng)歷日期,文中的示例代碼簡(jiǎn)潔易懂,有需要的小伙伴可以參考一下2024-11-11
六個(gè)實(shí)用Pandas數(shù)據(jù)處理代碼
這篇文章主要介紹了六個(gè)實(shí)用Pandas數(shù)據(jù)處理代碼，文章圍繞主題相相關(guān)內(nèi)容，具有一定的參考價(jià)價(jià)值，需要的小伙伴可以參考一下2022-05-05
這篇文章主要介紹了Python使用sorted排序的方法,結(jié)合三個(gè)實(shí)例分析了Python使用sorted方法進(jìn)行元素排序操作的相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2017-07-07
這篇文章主要介紹了Python使用裝飾器模擬用戶登陸驗(yàn)證功能,結(jié)合登錄驗(yàn)證實(shí)例形式分析了裝飾器的簡(jiǎn)單使用技巧,需要的朋友可以參考下2018-08-08
python?os.stat()如何獲取相關(guān)文件的系統(tǒng)狀態(tài)信息
這篇文章主要介紹了python?os.stat()如何獲取相關(guān)文件的系統(tǒng)狀態(tài)信息，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-11-11
這篇文章主要為大家介紹了Django視圖層與模板層實(shí)例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪<BR>2022-09-09
Python attrs提高面向?qū)ο缶幊绦试敿?xì)
Python是面向?qū)ο蟮恼Z(yǔ)言，一般情況下使用面向?qū)ο缶幊虝?huì)使得開發(fā)效率更高，軟件質(zhì)量更好，并且代碼更易于擴(kuò)展，可讀性和可維護(hù)性也更高，但是Python的類寫起來(lái)是真的累，這是可以在創(chuàng)建類的時(shí)候自動(dòng)添加上attrs模塊，下面文章我們就來(lái)介紹這個(gè)東西,需要的朋友可參考一下2021-09-09
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)八大排序算法的第一篇，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2017-09-09