pytorch环境搭建,mac环境
第一步安装Anconda
从anconda官网下载,不过官网太慢,建议去清华镜像源下载
下载链接: 安装包.
选择相应的安装包即可,如图所示
楼主是mac版本的,所以选择这个。windows(如果你的windows是64位的)中有选择x86_64的版本即可,下载安装(windows中有一步是需要添加conda环境到系统变量的选项,勾上即可)。
第二步配置镜像源,或者科学上网
下载安装好,打开anaconda的navigator(windows用户可能第一次闪退,需要右键以管理员省份运行),由于conda的镜像源在国外,所以需要配置相关镜像。
点击上图中的channel。
接着点击add,将如下的链接每个添加即可,可以删除default选项(其实也可以不删除),添加如上图所示,添加完成之后需要点击updata channels。
三个镜像源链接如下图所示:
http://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
http://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
第三步配置
接着进入pytorch官网,链接: 官网.
到开之后选取下图所示的osx即可,(windows)不带gpu的也就是没有cuda可以选择cpu only 那一行。
在mac的终端输入(windows中的cmd)该命令
conda install pytorch==1.6.0 torchvision==0.7.0 -c pytorch
即可,windows(conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch
)如下图所示,
等待程序执行即可,一般可能会报http error,多试几次就好了,如果还是不行建议晚上安装,或者科学上网(channel只保留default即可),安装成功之后,创建py文件中写入如下代码:
import torch
print(torch.__version__)
print('gpu:', torch.cuda.is_available())
运行成功即可查看torch的版本,以及gpu是否可运行。
如上图所示