Pytorch安装历程

1.首先需要安装一个anaconda3(可以到清华镜像中去下载,方便)

2.创建虚拟环境

为了避免与电脑中已存在的环境中发生冲突,新建一个虚拟环境,具体步骤参见https://blog.csdn.net/lyy14011305/article/details/59500819?ticket=ST-54761-uVouyW9tXaHgviR7YENY-passport.csdn.net,打开Anaconda Prompt 进行以下操作,例如:conda create -n torch python=3.6是创建一个名称为torch,python是3.6版本的虚拟环境;然后激活activate torch

3.安装Pytorch

https://pytorch.org/在官网选择与安装电脑匹配的版本,运行对应的语句

其中需要知道电脑的CUDA版本,参见https://www.cnblogs.com/yingchuan-hxf/p/7710071.html

安装torchvision
pip install torchvision

4.验证是否安装成功

输入python进入python环境,然后输入

>>>import torch

没有出现错误,就安装成功了!

猜你喜欢

转载自blog.csdn.net/qq_40279508/article/details/81078908