网络编程
位置:首页>> 网络编程>> Python编程>> Win10系统下Pytorch环境的搭建过程

Win10系统下Pytorch环境的搭建过程

作者:TFFboys  发布时间:2023-09-29 06:57:54 

标签:Win10,搭建,Pytorch,环境

前言

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

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

Win10系统下Pytorch环境的搭建过程

一、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环境的搭建过程

来源:https://blog.csdn.net/qq_36622589/article/details/117885123

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com