Pytorch環境搭建(Anaconda+Pycharm,清華鏡像源)


Pytorch環境搭建(Anaconda+Pycharm,清華鏡像源)

1. 安裝anaconda

前往Anaconda官網下載個人版
Image.png

2. 添加清華鏡像源

由於pytorch的服務器在國外,直連下載的話很慢,所以選用清華鏡像源下載

  1. 點擊開始菜單,進入Anaconda Prompt命令行
    https://img2020.cnblogs.com/news/1921416/202101/1921416-20210113223809079-556420723.jpg2. 在命令行中輸入以下命令
# 添加清華鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 添加pytorch鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes

3. 創建新的conda虛擬環境

conda create --name pytorch python=3.7
# 激活 進入新環境
activate pytorch

4. 安裝Pytorch

conda install pytorch torchvision cudatoolkit=10.2

這里比官網的命令少了一個-c pytorch
https://img2020.cnblogs.com/blog/1921416/202101/1921416-20210113223809407-59152343.jpg
主要是因為-c pytorch表示指定使用pytorch channel,這個服務器位於國外,下載速度很慢,使用剛剛添加進default channel的清華鏡像源可以得到一個較快的下載速度

驗證是否安裝成功在當前環境下輸入命令
python 
import torch 
import torchvision

5. 配置Pycharm

5.1 安裝Pycharm

前往Pycharm官網下載社區版,然后安裝

5.2 設置Python Interpreter

  1. 在新建project界面或是在主界面菜單File->Setting->Python Interpreter

  2. 設置Interpreter解釋器,選擇Existing Interpreter 已存在的解釋器
    https://img2020.cnblogs.com/blog/1921416/202101/1921416-20210113223809739-932438249.jpg

  3. 點擊三個點的按鈕去選擇解釋器的路徑
    https://img2020.cnblogs.com/news/1921416/202101/1921416-20210113223810125-627663239.jpg

  4. 選擇Conda Environment,設置解釋器路徑
    https://img2020.cnblogs.com/blog/1921416/202101/1921416-20210113223810391-1264169815.jpg

  5. 選擇Anaconda->envs->pytorch(剛剛創建的conda虛擬環境)->python.exe作為解釋器
    https://img2020.cnblogs.com/blog/1921416/202101/1921416-20210113223810695-1351608354.jpg

Pytorch在Pycharm的環境就搭建完成了


免責聲明!

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



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