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