手把手教你快速安裝gpu版本的pytorch(詳細(xì)圖文教程)
最近在win10裝torch，發(fā)現(xiàn)沒有Linux那么便捷，用conda和pip裝默認(rèn)版本都是cpu，而且下載速度很慢。網(wǎng)上的解決方法有好幾種，但是初學(xué)者可能面臨的問題是：我的系統(tǒng)該裝哪個(gè)版本的？cuda、torch、torchvision以及 torchaudio裝哪個(gè)版本？以及裝了cpu版本的顯示torch.cuda.is_available() False 該怎么辦？
那么，方法如下（親測(cè)成功好用）
第一步：創(chuàng)建環(huán)境 （-n +環(huán)境名（torch) + python版本（3.10））
conda create -n torch python=3.8
第二步：了解你電腦的cuda 版本，在terminal中輸入：nvidia-smi 即可，如下圖所示，版本為12.3
第三步：去pytorch官網(wǎng)看你的conda適配的torch、torchvision以及 torchaudio版本，這個(gè)鏈接：
Previous PyTorch Versions | PyTorch
比你的CUDA低版本的都能安裝，看好對(duì)應(yīng)的版本就可以了，不用官網(wǎng)的命令安裝（會(huì)很慢）
第四步：開始安裝，鑒于使用conda的各種問題，我們這里不使用conda安裝，使用pip安裝，根據(jù)剛才看好的版本，在創(chuàng)建的torch環(huán)境下輸入命令：
#進(jìn)入torch環(huán)境 conda activate torch
#用鏡像源快速安裝對(duì)應(yīng)版本 pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
第五步：驗(yàn)證torch安裝是否成功，import torch成功說明安裝成功，如果torch.cuda.is_available()返回False 時(shí)，說明裝了cpu版本的。不要慌，往下看。
第五步：查看torch環(huán)境下所有庫(kù)：
conda list
可以看到，pytorch是cpu版本的
第六步：手動(dòng)去清華源下載對(duì)應(yīng)的gpu版pytorch，和上面圖上看到的cpu的版本保持一致即可
Index of /anaconda/cloud/pytorch/ | 清華大學(xué)開源軟件鏡像站 | Tsinghua Open Source Mirror
比如pytorch版本是1.10.1（需要對(duì)應(yīng)），環(huán)境創(chuàng)建時(shí)python版本是3.8 （需要對(duì)應(yīng)），cuda 12.3（比它小的都可以），那么下面幾個(gè)可以下載，選一個(gè)下載，最好裝到一個(gè)英文路徑下。
第七步：離線安裝剛剛下載的文件
# cd 到下載的文件目錄，例如安裝目錄是E:\Download cd E:\Download
#離線安裝 conda install --offline pytorch-1.10.0-py3.6_cuda10.2_cudnn7.6.5_0.tar.bz2
最后一步： conda list, 查看pytorch 是否已經(jīng)被替換。可以看到替換成GPU版本了
再輸入 torch.cuda.is_available() 驗(yàn)證，返回True .大功告成！
總結(jié)
到此這篇關(guān)于快速安裝gpu版本的pytorch的文章就介紹到這了,更多相關(guān)快速安裝gpu版本pytorch內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Jupyter?notebook無法鏈接內(nèi)核、運(yùn)行代碼問題
文章主要介紹了在VSCode中使用Jupyter?Notebook遇到的問題及其解決過程,問題包括包版本沖突、文件沖突、路徑錯(cuò)誤和找不到文件,通過逐一排查和安裝相關(guān)依賴包,最終解決了這些問題,使得Jupyter?Notebook可以在VSCode中正常運(yùn)行2025-02-02
使用jupyter notebook直接打開.md格式的文件
這篇文章主要介紹了使用jupyter notebook直接打開.md格式的文件，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-04-04
Python利用wxPython模塊打造ChatGPT式打字效果程序
這篇文章主要為大家介紹了如何利用Python和wxPython模塊打造一個(gè)ChatGPT式打字效果程序，從而增強(qiáng)用戶體驗(yàn)或提高應(yīng)用程序的可讀性，感興趣的可以了解一下2023-05-05
tensorflow基于CNN實(shí)戰(zhàn)mnist手寫識(shí)別(小白必看)
這篇文章主要介紹了tensorflow基于CNN實(shí)戰(zhàn)mnist手寫識(shí)別(小白必看),文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
Python如何使用argparse模塊處理命令行參數(shù)
這篇文章主要介紹了Python如何使用argparse模塊處理命令行參數(shù),文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了Python基礎(chǔ)詳解之列表復(fù)制,文中有非常詳細(xì)的代碼示例,對(duì)正在學(xué)習(xí)python基礎(chǔ)的小伙伴們有很好的幫助,需要的朋友可以參考下2021-04-04
在軟件開發(fā)過程中,IPO（Input-Process-Output）分析是一種常用的方法,用于分析軟件系統(tǒng)的輸入、處理和輸出,Python作為一種功能強(qiáng)大的編程語言,提供了許多工具和庫(kù)來實(shí)現(xiàn)IPO分析,本文將介紹如何使用Python進(jìn)行IPO分析的基本步驟和常用工具,需要的朋友可以參考下2023-12-12
詳解Pandas如何高效對(duì)比處理DataFrame的兩列數(shù)據(jù)
我們?cè)谟?pandas?處理數(shù)據(jù)的時(shí)候，經(jīng)常會(huì)遇到用其中一列數(shù)據(jù)替換另一列數(shù)據(jù)的場(chǎng)景。這一類的需求估計(jì)很多人都遇到，當(dāng)然還有其它更復(fù)雜的。解決這類需求的辦法有很多，這里我們來推薦幾個(gè)2022-09-09
python re模塊findall()函數(shù)實(shí)例解析
這篇文章主要介紹了python re模塊findall()函數(shù)實(shí)例解析，具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01
Python的Flask框架中的Jinja2模板引擎學(xué)習(xí)教程
這篇文章主要介紹了Python的Flask框架中的Jinja2模板引擎學(xué)習(xí)教程,Jinja2模板引擎的用法也是Flask的Web開發(fā)中的重要知識(shí),需要的朋友可以參考下2016-06-06