win10搭建pytorch环境记录


前言

以下述环境为例:
python=3.7
pytorch=1.7.1
torchvision==0.8.2
cudatoolkit=10.1


pytorch和cuda版本对应参考下图:

在这里插入图片描述

一、anaconda安装

做机器学习和深度学习离不开环境和包管理,要养成每个工程单独设置环境的习惯!!!Anaconda和Miniconda两个自选(推荐前者,后者可以用在笔记本上节省空间)。
anaconda
链接: link.
miniconda
链接: link.

二、pytorch安装

1.新建环境

新版Anaconda现在默认python为最新版本,所以需要哪个版本得自己设置,常用新建环境命令在命令行中为:

conda create -n pytorch python=3.7 -y

#创建一个名为pytorch的环境,python版本3.7,确认

2.激活环境

activate pytorch

3.安装pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c pytorch

#该命令可以到官网 https://pytorch.org/get-started/locally/ 查看,选择对应选项,然后将run this command复制到命令行中。

在这里插入图片描述

总结

步骤很简单,但是过程中难免遇到各种问题,尤其是下载中进度不动或者中断了,重复操作即可,换源也是一种办法。

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值