Pytorch環境搭建(Anaconda+Pycharm,清華鏡像源)
1. 安裝anaconda
前往Anaconda官網下載個人版
2. 添加清華鏡像源
由於pytorch的服務器在國外,直連下載的話很慢,所以選用清華鏡像源下載
# 添加清華鏡像
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
主要是因為-c pytorch
表示指定使用pytorch channel,這個服務器位於國外,下載速度很慢,使用剛剛添加進default channel的清華鏡像源可以得到一個較快的下載速度
驗證是否安裝成功在當前環境下輸入命令
python
import torch
import torchvision
5. 配置Pycharm
5.1 安裝Pycharm
前往Pycharm官網下載社區版,然后安裝
5.2 設置Python Interpreter
-
在新建project界面或是在主界面菜單File->Setting->Python Interpreter
Pytorch在Pycharm的環境就搭建完成了