Pytorch环境的搭建(windows系统)

1.安装Anaconda

清华镜像源Index of /anaconda/archive/ | 北京外国语大学开源软件镜像站,致力于为国内和校内用户提供高质量的开源软件镜像、Linux 镜像源服务,帮助用户更方便地获取开源软件。本镜像站由清华大学 TUNA 协会负责运行维护。https://mirrors.bfsu.edu.cn/anaconda/archive/使用清华镜像源下载安装包,也可以去官方网站下载,(推荐使用镜像源下载,下载速度快)

双击此程序,安装anaconda。

2.环境配置

1.安装完成后,点击开始菜单,进入Anaconda Prompt命令行

2.在命令窗口添加下面命令

conda create --n pytorch python=3.6#创建新的环境
pytorch为环境名字,python=**可以甚至需要的版本。


 出现上面的选择,输入Y.。然后等待库的安装。

3.根据提示激活环境。

activate pytorch

4.添加镜像源命令

# 添加清华镜像
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/
conda config --set show_channel_urls yes

5.安装pytorch

进入pytorch官网,选择适合自己版本的conda(点击橘色字体)。

 因为我自己电脑版本为11.2,所以我下载的11.2版本。win+r输入cmd打开命令窗口,输入”nvidia-smi“查看自己的版本。

conda install pytorch torchvision cudatoolkit=11.2 

主要是因为-c pytorch表示指定使用pytorch channel,这个服务器位于国外,下载速度很慢,所以去掉了-c pytorch。

6.验证

1.python #会出现python版本,进入python内

2.import torch
3.torch.cuda.is_available()



依次输入上述命令,若最终返回True ,表示安装成功。

输入exit()退出即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值