1 前言

這兩天由于要使用Genesis，需要用到pytorch的環(huán)境，就裝了一下pytorch，記錄一下過(guò)程和踩坑記錄。因?yàn)殡娔X上要使用ArcGIS，因此默認(rèn)的python環(huán)境是不能改的，否則ArcGIS就不能啟動(dòng)了。因此新的python的環(huán)境需要使用Anaconda進(jìn)行創(chuàng)建，本文使用的環(huán)境包括：

• CUDA: 12.4
• Anaconda：2022.10
• Python：3.9
• Pytorch：2.5.1+cu124

2 安裝Anaconda

到官網(wǎng)https://www.anaconda.com/download 下載 windows 的安裝包直接安裝即可。

3 安裝CUDA

到CUDA官網(wǎng)下載安裝包

4 創(chuàng)建Python3.9環(huán)境

先從開(kāi)始菜單里找到conda的cmd控制臺(tái)

執(zhí)行創(chuàng)建環(huán)境命令

conda create -n pytorch_env python=3.9

執(zhí)行切換環(huán)境命令

conda activate pytorch_env

5 安裝Pytorch環(huán)境

到Pytorch官網(wǎng) https://pytorch.org/get-started/locally 選擇自己的環(huán)境，獲取相應(yīng)命令

5.1 conda方式

conda環(huán)境安裝命令為：

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

5.2 pip方式

pip環(huán)境安裝命令為：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

6 驗(yàn)證是否安裝成功

執(zhí)行以下命令：

import torchtorch.cuda.is_available()

返回True就表示安裝成功了。

7 注意事項(xiàng)

7.1 no module named torch問(wèn)題

說(shuō)明Pytorch安裝失敗了，連模塊都不能導(dǎo)入，檢查相應(yīng)版本后重新安裝。

7.12 torch.cuda.is_available()返回False問(wèn)題

能返回false，說(shuō)明Pytorch安裝成功了，但是版本不兼容。執(zhí)行一下命令看看是否安裝成了cpu版本

conda list

這就表示安裝成了cpu版本，而這也是使用conda安裝時(shí)默認(rèn)的版本，作者一開(kāi)始也是使用conda的命令安裝的，反復(fù)試了很多次都沒(méi)有成功，最后換成pip安裝，一次性就成功了。使用pip安裝后我們?cè)俅尾榭戳斜?，發(fā)現(xiàn)版本變了，這才正確版本。

8 最佳實(shí)踐

由于使用pip方式安裝時(shí)下載速度很慢，我們可以使用迅雷先將whl文件下載下來(lái)，然后通過(guò)pip命令安裝這個(gè)文件。當(dāng)我們使用pip命令時(shí)會(huì)發(fā)現(xiàn)工具會(huì)去下載一個(gè)whl文件

我這里因?yàn)橄螺d過(guò)了，地址變化了，之前的地址是

https://download.pytorch.org/whl/cu124/torch-2.5.1%2Bcu124-cp39-cp39-win_amd64.whl

我下載到了D盤(pán)，然后通過(guò)pip安裝，注意：不要有中文路徑

pip install D:\\torch-2.5.1+cu124-cp39-cp39-win_amd64.whl

9 總結(jié)

本文詳細(xì)的講解了Pytorch的安裝過(guò)程，以及中間遇到各種坑，更多相關(guān)pytorch安裝踩坑內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論