深度學習第一步 Windows+NVIDA配置 CUDA pytorch 環境
阿新 • • 發佈:2022-04-03
深度學習第一步 Windows+NVIDA配置 CUDA pytorch 環境
感謝
Windows 下安裝 CUDA 和 Pytorch 跑深度學習 - 動手學深度學習v2_嗶哩嗶哩_bilibili
目錄- 深度學習第一步 Windows+NVIDA配置 CUDA pytorch 環境
- 感謝
- 1 檢視電腦的GPU型別
- 2 cuda 下載與安裝
- 3 安裝anaconda
- 4 在anaconda下建立一個環境
- 5 下載Pytorch
- 6 安裝完成,檢驗
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
在啟用的環境中輸入命令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113