快捷導(dǎo)航

pytorch通過(guò)miniconda安裝的實(shí)現(xiàn)示例

更新時(shí)間：2025年06月03日 11:18:10 作者：xiaokinL

本文主要介紹了pytorch通過(guò)miniconda安裝的實(shí)現(xiàn)示例,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

最近跟著李沐老師在學(xué)習(xí)動(dòng)手深度學(xué)習(xí)，所以安裝一下pytorch環(huán)境，之前自己通過(guò)conda安裝過(guò)一次，新建了虛擬環(huán)境，踩了很多坑，這次想直接在原來(lái)的基礎(chǔ)上繼續(xù)用，結(jié)果d2l這個(gè)包一直裝不上，索性直接全刪了，重新再來(lái)一遍，剛好C盤(pán)被這個(gè)快撐爆了。

1. 確認(rèn)有NVIDIA GPU

桌面右鍵查看更多選項(xiàng)—NVIDIA控制面板，可以看到我這個(gè)是垃圾的MX 250

2、先給顯卡更新一下驅(qū)動(dòng)

這樣支持的cuda版本會(huì)多一點(diǎn)，

訪問(wèn)nVidia官網(wǎng)：人工智能計(jì)算領(lǐng)域的領(lǐng)導(dǎo)者 | NVIDIA，更新驅(qū)動(dòng)，點(diǎn)擊驅(qū)動(dòng)程序

找到自己的顯卡對(duì)應(yīng)的驅(qū)動(dòng)并下載：

下載完之后，打開(kāi)文件，一直點(diǎn)擊下一步，更新完驅(qū)動(dòng)。

3、查看現(xiàn)在支持的cuda版本

桌面右鍵查看更多選項(xiàng)—NVIDIA控制面板

可以看到現(xiàn)在支持的cuda版本最高為12.2.1

4、安裝CUDA

訪問(wèn)cuda下載：https://developer.nvidia.com/cuda-downloads

根據(jù)自己系統(tǒng)選擇，選擇本地下載

下載完成后，雙擊文件，會(huì)讓你選擇提取文件的文件夾，我選擇在D盤(pán)新建了一個(gè)文件夾，安裝完成之后，這個(gè)文件夾會(huì)被自動(dòng)刪除。

接下來(lái)，為了防止后面出問(wèn)題，我就選擇了精簡(jiǎn)安裝（C盤(pán)戰(zhàn)士，哭）

接下來(lái)等待即可，最后安裝完成。

查看是否安裝是否成功：cmd輸入nvidia-smi，注意高版本的CUDA這條指令沒(méi)有空格，李沐老師的安裝過(guò)程用的nvidia -smi，我試了半天都不對(duì)。

好像有種說(shuō)法是這里是指最高支持的CUDA版本，可以再輸入指令nvcc -V試一下

5、下載miniconda

選擇需要的python版本的miniconda ，如果沒(méi)有安裝過(guò)python的，可以隨便選，我選的python3.9版本，但是這個(gè)版本就算跟你自己的python不一樣也沒(méi)關(guān)系，反正后面選擇python解釋器的時(shí)候選這個(gè)就行。

下載好之后，安裝。

我選的安裝位置是D盤(pán)，因?yàn)镃盤(pán)不夠用了，如果C盤(pán)充足，最好放在盤(pán)，省的后面出現(xiàn)啥問(wèn)題，記住這個(gè)安裝路徑，后面需要用到。

注意，安裝路徑不能有中文，其余都是默認(rèn)選項(xiàng)

安裝完之后，找到anaconda powershell prompt(miniconda)

查看我們的python環(huán)境，確認(rèn)是3.9：

6、下載pytorch

在這個(gè)網(wǎng)站，或者直接搜pytorch

下拉找到下載：

Windows—pip安裝—python—選擇CUDA12.1（我的CUDA版本是12.2，可以向下兼容），根據(jù)自己情況選擇：

復(fù)制下方pip指令進(jìn)入剛才的miniconda的powershell中，防止下載過(guò)慢，可以在pip最后加一個(gè)中科大的鏡像源，-i https://pypi.mirrors.ustc.edu.cn/simple：

下載完成之后，看看torch能不能用，使用簡(jiǎn)單的張量運(yùn)算來(lái)檢查一下：

可以看到torch可以用，但是?。。。。。。。。。。。。。。。。。。。。。。。。。。。。?！

注意：檢查一下torch的版本，防止安裝成cpu版本（悲），torch.__version__

明明選的GPU版本，為啥安裝好了變成CPU版本了呢？？？解決方案可以看這個(gè)文章：

【精選】conda安裝GPU版pytorch，結(jié)果卻是cpu版本[找到問(wèn)題根源，從容解決]

我的解決方案是先切換conda安裝試一下，

在anaconda的powershell里試一下，如果出現(xiàn)提示，所有的包已經(jīng)存在，那就換下一個(gè)方法，如果直接重新安裝torch，等待安裝完再看看torch的版本。

上面的方法不行，就只能使用使用本地安裝，去conda官網(wǎng)：https://anaconda.org/，找到我要的pytorch安裝包：

下載后把這個(gè)bz2文件放到：D:\miniconda\pkgs這個(gè)文件夾里面，就是剛在安裝miniconda的哪個(gè)路徑，然后在anaconda的powershell里面，先切盤(pán)到D:\miniconda\pkgs

然后，用conda命令進(jìn)行本地安裝：

conda install --use-local pytorch-2.1.0-py3.9_cuda12.1_cudnn8_0.tar.bz2后面這一串就是文件名

現(xiàn)在看看torch版本，后面不帶cpu了，再試試能不能調(diào)用cuda，然后我的問(wèn)題就解決了。

蕪湖，起飛！

7.vscode運(yùn)行

最后，我一般用vscode，然后在vscode里面，先選擇python語(yǔ)言

然后在選擇解釋器，就是用我們剛才在miniconda里面安裝的python解釋器，因?yàn)槲业膍iniconda的安裝路徑是在D盤(pán)，所以找到D盤(pán)里的python程序，最后地址是D:/miniconda/python.exe，剛才的torch也在這里。

選好之后我們測(cè)試一下最簡(jiǎn)單的線性模型：

win！！

總結(jié)

多看看安裝教程，一定注意核對(duì)電腦支持的CDUA版本+python版本+pytorch版本，版本兼容問(wèn)題應(yīng)該是新手遇到的最大的問(wèn)題，其次，還有在conda中安裝的方法，也可以嘗試一下。

到此這篇關(guān)于pytorch通過(guò)miniconda安裝的實(shí)現(xiàn)示例的文章就介紹到這了,更多相關(guān)pytorch通過(guò)miniconda安裝內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: