這篇文章主要講解了“Window下Python+CUDA+PyTorch怎么安裝”，文中的講解內(nèi)容簡單清晰，易于學(xué)習(xí)與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學(xué)習(xí)“Window下Python+CUDA+PyTorch怎么安裝”吧！

公司主營業(yè)務(wù)：網(wǎng)站制作、成都網(wǎng)站制作、移動網(wǎng)站開發(fā)等業(yè)務(wù)。幫助企業(yè)客戶真正實現(xiàn)互聯(lián)網(wǎng)宣傳，提高企業(yè)的競爭能力。創(chuàng)新互聯(lián)建站是一支青春激揚、勤奮敬業(yè)、活力青春激揚、勤奮敬業(yè)、活力澎湃、和諧高效的團隊。公司秉承以“開放、自由、嚴謹、自律”為核心的企業(yè)文化，感謝他們對我們的高要求，感謝他們從不同領(lǐng)域給我們帶來的挑戰(zhàn)，讓我們激情的團隊有機會用頭腦與智慧不斷的給客戶帶來驚喜。創(chuàng)新互聯(lián)建站推出章丘免費做網(wǎng)站回饋大家。

1 概述

Windows下Python+CUDA+PyTorch安裝，步驟都很詳細，特此記錄下來，幫助讀者少走彎路。

2 Python

Python的安裝還是比較簡單的，從官網(wǎng)下載exe安裝包即可：

因為目前最新的torch版本只支持到Python 3.8，因此為了不會出現(xiàn)版本兼容問題建議安裝Python 3.8。

下載中間的那個executable installer并安裝，打開會建議選擇自定義：

選擇需要的組件：

選擇位置：

安裝完后可以從命令行查看是否安裝成功（一般默認會把路徑添加到環(huán)境變量中）：

python --version

由于該安裝包會自動把pip也一起安裝，因此在使用時會提示pip需要升級，請使用如下命令升級pip：

python -m pip install --upgrade pip

在安裝前建議添加國內(nèi)源，在用戶目錄下新建一個pip文件夾，并在其中新建一個叫pip.ini的文件，輸入如下內(nèi)容：

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = https://pypi.tuna.tsinghua.edu.cn

3 CUDA

CUDA的安裝也不難，一般Windows會在裝系統(tǒng)的時候自動安裝好對應(yīng)的N卡驅(qū)動，從官網(wǎng)下載對應(yīng)的CUDA Toolkit：

下載好后打開（實際上是一個壓縮包，這里是解壓路徑）：

解壓完成后會自動打開安裝界面：

沒特殊需求可以直接使用默認安裝選項，否則請選擇自定義組件進行安裝：

由于筆者沒有安裝VS，這里會提示找不到VS，忽略即可：

接下來就是等待安裝：

安裝完成：

打開cmd輸入nvcc --version，顯示版本號即為安裝成功：

4 PyTorch

4.1 下載

PyTorch建議使用離線方式手動安裝，戳這里下載對應(yīng)CUDA版本與對應(yīng)Python版本的torch以及torchvision：

最前面的cu110表示cuda 11.0，接著是torch/torchvision，再接著是版本，后面的cp表示Python版本，比如cp36表示Python 3.6，接著是系統(tǒng)和架構(gòu)。

4.2 安裝

下載完成后使用pip安裝即可：

pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl

注意如果是Windows 2004版本可能會出現(xiàn)如下問題：

這是numpy 19.0.4版本的問題，詳情可以戳這里。

解決方法是安裝19.0.3版本：

pip uninstall numpy
pip install numpy==19.0.3

接著進行測試，打開python：

python
import torch
print(torch.__version__)

輸出torch版本即表示成功：

感謝各位的閱讀，以上就是“Window下Python+CUDA+PyTorch怎么安裝”的內(nèi)容了，經(jīng)過本文的學(xué)習(xí)后，相信大家對Window下Python+CUDA+PyTorch怎么安裝這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是創(chuàng)新互聯(lián)，小編將為大家推送更多相關(guān)知識點的文章，歡迎關(guān)注！