使用requirements.txt文件安装cuda(GPU)版本的pytorch
创始人
2024-12-29 10:09:26
0

使用requirements.txt文件安装cuda(GPU)版本的pytorch

  • 问题描述
  • 解决方法

问题描述

使用requirements.txt可以轻松地帮助我们配置新环境,然而,当使用requirements.txt安装pytorch时有时会出现仅能安装cpu版本pytorch的情况。
举例说明:
requirements.txt文件内容如下:

torch==2.0.1 torchaudio==2.0.2 torchvision==0.15.2 

此时在anaconda promot中输入

pip install -r requirements.txt 

然后会发现,安装的是cpu版本的pytorch

在这里插入图片描述

这里输出的也是False
在这里插入图片描述
在这里插入图片描述

解决方法

应该将requirements.txt文件改为如下形式

torch==2.0.1+cu117 torchaudio==2.0.2+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html 

然后再在anaconda promot中输入

pip install -r requirements.txt 

此时安装的就是cuda版本的pytorch

在这里插入图片描述
这里输出是True
在这里插入图片描述
在这里插入图片描述

相关内容

热门资讯

透视能赢(WepoKe)外挂透... 1、透视能赢(WepoKe)外挂透明挂辅助app(wepoke真的有挂)素来存在有挂(详细透视技巧教...
aapoker发牌机制!aap... aapoker发牌机制!aapoker有外挂吗,(德州aapoker俱乐部)本来真的有挂(详细辅助技...
透视脚本(wepOke)透明挂... 这是一款非常优秀的wepoke的确有挂 ia辅助检测软件,能够让你了解到wepoke的确有挂中牌率当...
aa扑克辅助!德州aa扑克透明... aa扑克辅助!德州aa扑克透明挂,(aaPOKER)一直是真的有挂(详细辅助细节揭秘)1、首先打开a...
透视计算(wepOKE)透明挂... 透视计算(wepOKE)透明挂辅助代打(WePoKe外挂)切实真的是有挂(详细透视攻略教程);支持多...
aapoker外挂!aapok... aapoker外挂!aapoker外挂教程,(aApoker)总是是真的有挂(详细辅助细节揭秘)aa...
透视辅助(WepOke)外挂透... 《透视辅助(WepOke)外挂透明挂辅助软件(wepokeai辅助)切实是真的有挂(详细透视黑科技教...
aapoker辅助!aapok... aapoker辅助!aapoke辅助工具存在吗,(aa poker)好像有挂(详细辅助2025新版教...
透视数据(WEpoke)透视辅... 透视数据(WEpoke)透视辅助神器(wepokeai辅助)好像是有挂(详细透视分享教程);支持2-...
aapoker俱乐部!aa扑克... aapoker俱乐部!aa扑克有挂吗,(AApoker)都是真的有挂(详细辅助存在挂教程)1、玩家可...