Pytorch环境搭建(Anaconda+Pycharm,清华镜像源)

Pytorch环境搭建(Anaconda+Pycharm,清华镜像源)

1. 安装anaconda

前往Anaconda官网下载个人版

2. 添加清华镜像源

由于pytorch的服务器在国外,直连下载的话很慢,所以选用清华镜像源下载

  1. 点击开始菜单,进入Anaconda Prompt命令行
  2. 在命令行中输入以下命令
# 添加清华镜像
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
  1. 在新建project界面或是在主界面菜单File->Setting->Python Interpreter

  2. 设置Interpreter解释器,选择Existing Interpreter 已存在的解释器

  3. 点击三个点的按钮去选择解释器的路径

  4. 选择Conda Environment,设置解释器路径

  5. 选择Anaconda->envs->pytorch(刚刚创建的conda虚拟环境)->python.exe作为解释器

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

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值