Windows10系統下pytorch環境快速配置(Anaconda+CUDA+torch+PyCharm)


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環境下


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM