windows + anaconda + pycharm + nvidia + cuda + cudnn + pytorch 安裝教程
- 查看顯卡資訊
- 查看cuda版本
- 安裝cuda
- 安裝cudnn
- 配置cuda環境變數
- 配置cudnn環境變數
- 安裝pytorch
- conda安裝
- pip安裝
- 第一個DEMO
要學pytorch了耶，這不得安裝一把子環境？
查看顯卡資訊
操作：開始 —— 右鍵 —— 設備管理器 —— 顯示配接器，
查看cuda版本
操作：右下角NVIDIA —— 右鍵 —— NVIDIA控制面板 —— 系統資訊 —— 組件，
安裝cuda
去CUDA官網下載cuda10.2版本，
下載后雙擊exe檔案，更改目錄，解壓，
解壓完成后會自動進入安裝流程，
按照安裝提示默認安裝即可，
安裝中，
檢查是否安裝成功，
nvcc -V
成功嘞！
安裝cudnn
去CUDNN官網下載cuda10.2版本對應的cudnn，
下載之后，解壓縮，將CUDNN壓縮包里面的bin、include、lib檔案直接復制到CUDA的安裝目錄下（C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2），
這里會遇到管理員權限問題，
在”為所有當前專案執行此操作“前點擊對號再點擊繼續，
配置cuda環境變數
已經有兩個了，
還要添加三個，
一開始找不到CUDA Samples，但是開始選單可以看到，
打開即可，
后來發現，系統變數的PATH里有bin和libnvvp就可以了，
配置cudnn環境變數
安裝pytorch
在conda里面查看python版本，
在conda里面創建虛擬環境，
wxm_pytorch是虛擬環境名字，
conda create -n wxm_pytorch python=3.7
點擊y，
激活虛擬環境，
conda activate wxm_pytorch
進入pytorch官網選擇配置項，
conda安裝
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
輸入y，等待安裝，
下好啦，
下錯了，
先洗掉這個失敗的虛擬環境，
重來，
pip安裝
pip3 install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
下好了，
測驗一下：
開心心，
看到torch了，
第一個DEMO
終于！！！
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/437962.html
標籤：AI
上一篇：3天AI進階實戰營——多目標跟蹤
下一篇：R語言-嶺回歸的代碼與案例解讀