2024最新Pytorch安裝圖文教程(Anaconda+GPU)
一、Anaconda創(chuàng)建虛擬環(huán)境
二、CUDA安裝
1.查看本地顯卡驅(qū)動對應(yīng)CUDA版本號 首先找到NVIDIA控制面板
點擊左下角系統(tǒng)信息:
點擊組件:
選擇適合自己電腦的cuda版本,比上圖所支持的CUDA 版本低都行!
2.官網(wǎng)下載:
CUDA Toolkit Archive | NVIDIA Developer
雙擊下載后的exe文件:
由于我的電腦沒有Visual Studio,所以我選擇自定義安裝取消該選項。本人是初學,具體有什么影響暫時還不清楚,不過目前位置沒有影響使用。
3.查看系統(tǒng)環(huán)境變量
4.測試CUDA是否安裝成功
三、CUDNN安裝
cuDNN Archive | NVIDIA Developer
有的時候好像不需要創(chuàng)建賬戶就能下載。
在cudnn壓縮包下載路徑,首先解壓壓縮包,提高用戶使用這三個文件夾的權(quán)限:
將這三個文件復(fù)制到CUDA的安裝目錄下
四、Pytorch安裝
打開Anaconda終端
切換到上面創(chuàng)建的虛擬環(huán)境DeepLearning為最開始你創(chuàng)建的虛擬環(huán)境名稱。
conda activate DeepLearning
安裝:
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
遇到問題:conda安裝成了cpu版本
Conda list
可以看到,只能找到cpu版的。
所以卸載掉CPU版本的Pytorch包,用命令:conda uninstall pytorch
添加鏡像:
?conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
添加完鏡像后,可以找到GPU版的pytorch:
重新在官網(wǎng)https://pytorch.org/get-started/previous-versions/中找歷史版本pytorch的下載命令:
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
驗證:GPU版本安裝成功
到此這篇關(guān)于2024最新Pytorch安裝教程(Anaconda+GPU)的文章就介紹到這了,更多相關(guān)Pytorch安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
如何使用Python JSON解析和轉(zhuǎn)換數(shù)據(jù)
JSON 是文本,使用 JavaScript 對象表示法編寫,Python 有一個內(nèi)置的 json 包,可用于處理 JSON 數(shù)據(jù),本文給大家介紹使用Python JSON解析和轉(zhuǎn)換數(shù)據(jù)的方法,感興趣的朋友跟隨小編一起看看吧2023-11-11python基于socket函數(shù)實現(xiàn)端口掃描
這篇文章主要為大家詳細介紹了python基于socket函數(shù)實現(xiàn)端口掃描,文中示例代碼介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們可以參考一下2020-05-05Python中利用all()來優(yōu)化減少判斷的實例分析
在本篇文章里小編給大家整理的是一篇關(guān)于Python中利用all()來優(yōu)化減少判斷的實例分析內(nèi)容,有需要的朋友們可以學習下。2021-06-06Python中關(guān)鍵字nonlocal和global的聲明與解析
這篇文章主要給大家介紹了關(guān)于Python中關(guān)鍵字nonlocal和global的聲明與解析的相關(guān)資料,文中介紹的非常詳細,相信對大家具有一定的參考價值,需要的朋友們下面來一起看看吧。2017-03-03