打开APP
userphoto
未登录

开通VIP,畅享免费电子书等14项超值服

开通VIP
这两个有什么区别pip install torch==2.0.1+cu118 pip install torch==2.0.1
userphoto

2023.07.01 美国

关注
表白:黑白圣堂血天使,天剑鬼刀阿修罗。 
讲解对象:
/这两个有什么区别pip install torch==2.0.1+cu118 pip install torch==2.0.1
作者:融水公子 rsgz
===


这两个有什么区别
pip install torch==2.0.1+cu118
pip install torch==2.0.1

按道理来说一个制定了cu118  一个没有指定,肯定是有区别的,对的,如果你是这么思考的话,思路是对的

这两个命令的区别在于是否指定了编译环境。第一个命令指定了编译环境为 CU

DA 11.1,而第二个命令没有指定编译环境,因此默认使用 CPU 版本的 PyTorch。


如果你的机器支持 CUDA,并且你想要使用 GPU来进行加速计算,那么你需要安装
CUDA 对应的 PyTorch 版本,并
且在安装时指定正确的编译环境。

如果你不需要使用 GPU,或者你的机器不支持 CUDA,那么你可以使用 CPU 版本的
PyTorch。

需要注意的是,使用 GPU 版本的 PyTorch 需要满足一定的硬件和软件要求,例

如需要安装 NVIDIA 显卡驱动和 CUDA 工具包等。如果你不确定自己

的机器是否支持 CUDA 或者如何安装这些依赖项,建议先进行相关的调研和学

习。



=== 
公众号:不浪仙人
谢谢大家的支持!可以点击我的头像,进入我的空间浏览更多文章呢。建议大家360doc[www.360doc.com]注册一个账号登录,里面真的有很多优秀的文章,欢迎大家的到来。
---
本站仅提供存储服务,所有内容均由用户发布,如发现有害或侵权内容,请点击举报
打开APP,阅读全文并永久保存 查看更多类似文章
猜你喜欢
类似文章
【热】打开小程序,算一算2024你的财运
pytorch安装及卸载
linux环境下使用Conda 安装pytorch
win10+pytorch1.0+cuda9
从0到1深度学习环境搭建
详解PyTorch编译并调用自定义CUDA算子的三种方式
解决Linux下无法使用GPU,torch.cuda.is_available()返回False的问题(无论怎么下载都是cpu版本的问题)
更多类似文章 >>
生活服务
热点新闻
分享 收藏 导长图 关注 下载文章
绑定账号成功
后续可登录账号畅享VIP特权!
如果VIP功能使用有故障,
可点击这里联系客服!

联系客服