1. 程式人生 > 其它 >深度學習第一步 Windows+NVIDA配置 CUDA pytorch 環境

深度學習第一步 Windows+NVIDA配置 CUDA pytorch 環境

深度學習第一步 Windows+NVIDA配置 CUDA pytorch 環境

感謝

Windows 下安裝 CUDA 和 Pytorch 跑深度學習 - 動手學深度學習v2_嗶哩嗶哩_bilibili

目錄

1 檢視電腦的GPU型別

1 在 cmd命令列之中 輸入dxdiag / 在執行中shuru dxdiag

2 之後就會開啟 DirectX 診斷工具

在顯示一欄下面就有顯示卡資訊

2 cuda 下載與安裝

CUDA Toolkit 11.6 Update 2 Downloads | NVIDIA Developer

安裝選預設選項就可以,把驅動和開發環境選上

看一下系統環境配置

在命令列中輸入用nvidia-smi來看一下是否安裝成功

3 安裝anaconda

Anaconda | The World's Most Popular Data Science Platform

4 在anaconda下建立一個環境

點選綠色箭頭啟動環境

5 下載Pytorch

Start Locally | PyTorch

在啟用的環境中輸入命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

6 安裝完成,檢驗