0
点赞
收藏
分享

微信扫一扫

Anaconda下Pytorch环境配置

你的益达233 2022-04-18 阅读 185
python

Anaconda 安装

比较简单,官网下载,无脑一步一步进行安装即可

Anaconda 修改镜像

打开资源管理器,开启隐藏文件显示,找到C盘中的用户文件夹下的Anaconda配置文件.condarc:C:\Users\用户名.condarc,修改内容如下:

channels:
# pytorch的镜像
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# DGL库的镜像
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/dglteam/
# 其他包的下载镜像,,可以根据需要添加不同的镜像地址
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
show_channel_urls: true
sl_verify: false

Anaconda 创建虚拟环境

命令:conda create -n 环境名 python=3.x

如果不修改镜像,直接创建虚拟环境,可能会报错,类似:CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/python/win-64/current_repodata.json>

这个错误并不是一定会出现,如果出现,解决方案如下:

按照上述Anaconda 修改镜像中的描述修改镜像,再使用该命令进行安装即可。

使用官网命令能够成功安装且不报错,则不需要修改镜像。

Anaconda 安装 Pytorch 环境

Pytorch官网:https://pytorch.org/get-started/locally/

1、先创建虚拟环境

命令:conda create -n 环境名 python=3.x

实例:conda create -n pytorch python=3.9,即创建一个基于Python3.9版本且名称为 pytorch的虚拟环境

2、激活虚拟环境

命令:conda activate 环境名

3、安装 Pytorch

官网命令(CPU版本):conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU版本选择对应的CUDA版本即可。

如果Anaconda的镜像修改为了清华的镜像,则不能直接使用官网的命令,需要删去官网命令末尾的-c pytorch,如果不修改依然会报CondaHTTPError错误,新命令为:conda install pytorch torchvision torchaudio cpuonly

Anaconda 安装 DGL

DGL官网:https://www.dgl.ai/pages/start.html

DGL需要先安装Pytorch,流程如 Anaconda 安装 Pytorch 环境 所述。

DGL官网安装命令(CPU版本):conda install -c dglteam dgl

使用镜像之后也需要修改命令,删除-c dglteam,如果使用原命令,可能会报CondaHTTPError错误,新命令为:conda install dgl

如果需要安装GPU加速版本,只需要在新命令的基础上修改dgl对应的cuda版本名即可,例如conda install dgl-cuda113

举报

相关推荐

0 条评论