华域联盟 Python Win10系统下Pytorch环境的搭建过程

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复制到命令行中。

总结

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

到此这篇关于Win10系统Pytorch环境搭建过程的文章就介绍到这了,更多相关Win10搭建Pytorch环境内容请搜索华域联盟以前的文章或继续浏览下面的相关文章希望大家以后多多支持华域联盟!

您可能感兴趣的文章:

本文由 华域联盟 原创撰写:华域联盟 » Win10系统下Pytorch环境的搭建过程

转载请保留出处和原文链接:https://www.cnhackhy.com/28724.htm

本文来自网络,不代表华域联盟立场,转载请注明出处。

作者: sterben

发表回复

联系我们

联系我们

2551209778

在线咨询: QQ交谈

邮箱: [email protected]

工作时间:周一至周五,9:00-17:30,节假日休息

关注微信
微信扫一扫关注我们

微信扫一扫关注我们

关注微博
返回顶部