1. Anaconda官網下載並安裝
https://www.anaconda.com/products/individual
2. 打開Anaconda Prompt,創建虛擬環境
conda create -n torch python=3.8 (python對應步驟1中版本)
激活環境:
conda activate torch
此處是為了防止配置多個環境時發生沖突。
3. CUDA官網下載Toolkit
https://developer.nvidia.com/cuda-toolkit-archive
此處可先查閱一下電腦顯卡對應的CUDA版本:
步驟:打開NVIDIA控制面板左下角點擊系統信息;
切換到組件的位置可以看到顯卡支持的CUDA版本;
4. Pytorch官網配置torch
此處先連接到清華鏡像,將加速下載,具體命令為:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后進入Pytorch官網 https://pytorch.org/get-started/locally/,下載對應版本的torch
5. 在命令行中測試配置結果
首先,激活torch虛擬環境,conda activate torch;
然后,輸入python,進入到python環境中;
最后,import torch,看torch環境是否已經配置成功。
6. PyCharm官網下載
https://www.jetbrains.com/pycharm/download/#section=windows
如果是學生用戶,可下載專業版,然后按照官網步驟用edu郵箱申請,時間為一年,過期后再次申請即可;
如果不是學生,直接下載社區版即可。
7. pycharm中pytorch環境配置,選擇到上述步驟中的虛擬torch環境下