使用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
在这里插入图片描述
在这里插入图片描述

相关内容

热门资讯

透视插件!竞技联盟辅助,wep... 透视插件!竞技联盟辅助,wepoker破解器激活码,窍要教程(有挂教程)-哔哩哔哩1、游戏颠覆性的策...
第一分钟app!胡乐辅助脚本,... 第一分钟app!胡乐辅助脚本,潮汕汇辅助神器,总是存在有挂(真实有挂)-哔哩哔哩暗藏猫腻,小编详细说...
透视美元局!德普辅助器怎么用,... 透视美元局!德普辅助器怎么用,wepoker插件辅助,指引教程(新版有挂)-哔哩哔哩该软件可以轻松地...
9分钟攻略!花花生活圈脚本,大... 9分钟攻略!花花生活圈脚本,大唐撸麻雀辅助码,一直真的有挂(有挂解密)-哔哩哔哩1、大唐撸麻雀辅助码...
透视了解!哈糖大菠萝免费辅助器... 透视了解!哈糖大菠萝免费辅助器,hardrock辅助,妙计教程(真是有挂)-哔哩哔哩1、进入到哈糖大...
第6分钟方法!丽水都莱辅助器,... 第6分钟方法!丽水都莱辅助器,榕城510k辅助软件,真是存在有挂(今日头条)-哔哩哔哩榕城510k辅...
透视真的!wepokerplu... 透视真的!wepokerplus透视脚本免费,wepoker轻量版透视系统,项目教程(有挂神器)-哔...
五分钟插件!皮皮四川游戏挂,途... 五分钟插件!皮皮四川游戏挂,途游大作战辅助,一贯真的是有挂(有挂方式)-哔哩哔哩1、玩家可以在途游大...
透视脚本!poker worl... 透视脚本!poker world辅助,hhpoker软件安装包,教程书教程(有挂教学)-哔哩哔哩透视...
第六分钟开挂!心悦透视辅助软件... 第六分钟开挂!心悦透视辅助软件,越乡游义乌辅助器微信免费,必备教程-2026最新版本越乡游义乌辅助器...