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

相关内容

热门资讯

辅助一下!微信小程序填大坑辅助... 辅助一下!微信小程序填大坑辅助器,广西老友玩辅助,果然存在有挂(哔哩哔哩)1、游戏颠覆性的策略玩法,...
解谜一下!悟空大厅辅助,决战手... 解谜一下!悟空大厅辅助,决战手游辅助,竟然是真的有挂(哔哩哔哩)亲,关键说明,悟空大厅辅助透视脚本安...
曝光一下!天天微友有辅助麻,战... 曝光一下!天天微友有辅助麻,战神辅助直装破解版,果然真的是有挂(哔哩哔哩)1、起透看视 天天微友有辅...
科普一下!奇迹山西辅助器,游戏... 科普一下!奇迹山西辅助器,游戏茶苑辅助,一贯真的是有挂(哔哩哔哩)1、打开软件启动之后找到中间准星的...
总结一下!微信小程序家乡大贰破... 总结一下!微信小程序家乡大贰破解版,乐乐休闲游戏辅助,一贯真的有挂(哔哩哔哩)微信小程序家乡大贰破解...
普及一下!西兵互娱辅助插件ap... 普及一下!西兵互娱辅助插件app,新上游拼十辅助,果然真的有挂(哔哩哔哩)1、进入游戏-大厅左侧-新...
教你一下!雀友会免费自动挂机脚... 教你一下!雀友会免费自动挂机脚本,大懒人斗十四辅助,一直存在有挂(哔哩哔哩)1、完成雀友会免费自动挂...
辅助一下!哈局透视辅助,新玄龙... 辅助一下!哈局透视辅助,新玄龙插件,其实有挂(哔哩哔哩)该软件可以轻松地帮助玩家将哈局透视辅助外卦神...
揭露一下!吉林心悦游戏辅助,八... 揭露一下!吉林心悦游戏辅助,八仙竞技馆辅助器下载,其实是有挂(哔哩哔哩);在进入吉林心悦游戏辅助软件...
了解一下!佛手在线大菠萝可以作... 了解一下!佛手在线大菠萝可以作弊码,新星游辅助软件下载,本来真的有挂(哔哩哔哩)1、在佛手在线大菠萝...