当前位置: 首页 > news >正文

安装pytorch

一、在anaconda中创建虚拟环境

打开Anaconda Prompt创建一个虚拟环境。比如要创建一个名字为pytorch的虚拟环境,可以如下输入。其中python=3.7指定该虚拟环境的python版本号。

conda create -n pytorch python=3.7

二、进入新创建的虚拟环境。

创建好虚拟环境后,通过在Anaconda Prompt中输入conda activate pytorch进入刚刚创建的虚拟环境pytorch。

conda activate pytorch

三、安装pytorch

进入pytorch官网(PyTorch),在首页下拉,如下图所示,选择相应的版本,将下图中我用箭头指出的红色方框里的内容输入Anaconda Prompt的命令行中即可开始pytorch的下载与安装。如果自己电脑的显卡支持cuda,则选择相应版本号的cuda,否则选择CPU版本。

 

四、确定是否安装成功

首先输入python进入python环境中。

然后输入import python,如果没有报错说明pytorch安装成功。

接着输入torch.cuda.is_available(),如果返回true,说明这台机器的GPU已经可以被pytorch使用。

 

五、故障说明

1.输入安装pytorch的命令后无反应:即输入安装命令后卡在某一步而不进行pytorch的下载与安装。这种情况建议更换其它版本号的cuda。

2.pytorch安装成功,但torch.cuda.is_available()返回false:这是也建议更换其它的版本。下面两张图是我两次安装pytorch时的情况,其中第一张图最后torch.cuda.is_available()输出的是false,第二张图输出的是true。从两张图中我用红框标出的内容也可以看出第一张图输出false的原因。之所以第二张图能成功是因为我换了版本进行安装。在安装pytorch时会像下图一样提示你将会安装哪些包,在你输入y后放会进行安装,可以看图中我用红框框起来的内容决定是否安装。

 

 

 

http://www.lryc.cn/news/26075.html

相关文章:

  • 自动化测试 Appium之Python运行环境搭建 Part2
  • LeetCode 2 - 两数相加
  • 用Python实现九九乘法表的几种方式,最简单只需一行代码
  • 入职外包三个月,我提桶跑路了
  • 企业邮箱对企业有哪些好处以及便捷性
  • ConvNeXt V2实战:使用ConvNeXt V2实现图像分类任务(一)
  • 3.2 报错整理
  • 从0开始学python -46
  • JavaScript事件委托机制详解
  • 【项目实战】MySQL中union和union all的相同点与不同点
  • ChatGPT最牛应用,让它帮你更新网站新闻吧!
  • 乌班图安装kvm并配置网络
  • 蓝库云|ERP系统在企业数字化转型中最常用的八大功能
  • Pytorch学习笔记#1:拟合函数/梯度下降
  • 挑战图像处理100问(24)——伽玛校正
  • 高级信息系统项目管理师(高项)软考论文评分标准(附历年高项论文题目汇总)
  • MySQL实战记录篇2
  • C++实现AVL树
  • 高并发语言erlang编程初步
  • springboot 问题记录
  • 【PAT甲级题解记录】1034 Head of a Gang (30 分)
  • Python搭建一个steam钓鱼网站,只要免费领游戏,一钓一个准
  • maven 私服nexus安装与使用
  • 详解数据结构中的顺序表的手动实现,顺序表功能接口【数据结构】
  • 【二】kubernetes操作
  • 如何在 C++ 中调用 python 解析器来执行 python 代码(五)?
  • 八 SpringMVC【拦截器】登录验证
  • PhotoShop基础使用
  • 借助阿里云 AHPA,苏打智能轻松实现降本增效
  • 美团2面:如何保障 MySQL 和 Redis 数据一致性?这样答,让面试官爱到 死去活来