Windows10系统 Anaconda创建pytorch环境 过程笔记

2021/9/6 7:08:20

本文主要是介绍Windows10系统 Anaconda创建pytorch环境 过程笔记,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

安装pytorch之前需要已经安装好Anaconda:Anaconda安装笔记

1、首先打开Anaconda Prompt。开始->Anaconda3->Anaconda Prompt(以管理员的身份运行)

2、查看现在环境列表:

conda env list

3、创建名为“pytorch”的屋子,python版本指定为3.6(中途要输入y)

conda create -n pytorch python=3.6

4、查看现在的列表:

5、激活并进入这个环境

conda activate pytorch

***************************************************************准备工作****************************************************************

查看自己显卡的版本,不会的可以看一下教程,我的版本如下:

从网站查看适合自己GPU版本的CUDA版本:

 

打开官网,PyTorch Build 选择 Stable;选择系统Windows;Package,Windows下推荐 conda,Linux 下推荐 pip;language选择Python;Compute Platform,如果有显卡选择GPU版本,否则CPU。

提示:发现推荐的CUDA版本是10.2,而我适合的是9.0版本,10.2可能会出现问题。

于是通过网页来选择安装合适的pytorch版本:

接下来复制命令

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

*************************************************************准备工作结束**************************************************************

6、回到刚才的Anaconda Prompt,输入刚才复制的命令(过程中需要输入“y”):

7、安装结束之后,验证是否安装成功:

import torch
torch.cuda.is_available()

8、退出python可用Ctrl+Z敲回车,退出pytorch环境用deactivate。

 



这篇关于Windows10系统 Anaconda创建pytorch环境 过程笔记的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程