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

相关内容

热门资讯

详情透视!wepoker黑侠破... 详情透视!wepoker黑侠破解,微乐智能辅助app,指南教程(有挂详情)-哔哩哔哩1.wepoke...
针对!柳州天天爱字牌辅助器,青... 针对!柳州天天爱字牌辅助器,青橙竞技游戏辅助,方案教程(有挂透明挂)-哔哩哔哩1、许多玩家不知道柳州...
突发!拱趴大菠萝机器人,熟客温... 突发!拱趴大菠萝机器人,熟客温州游戏辅助器,模板教程(有挂神器)-哔哩哔哩1、熟客温州游戏辅助器脚本...
科普透视!wpk作弊,微信小程... 科普透视!wpk作弊,微信小程序透视微乐,法子教程(有挂教程)-哔哩哔哩亲,关键说明,wpk作弊透视...
2026版教学!哈糖大菠萝攻略... 2026版教学!哈糖大菠萝攻略,新星游辅助软件,方式教程(有挂工具)-哔哩哔哩1、哈糖大菠萝攻略透视...
据统计!新上游辅助,拱趴大菠萝... 据统计!新上游辅助,拱趴大菠萝辅助工具是真的吗,教材教程(揭秘有挂)-哔哩哔哩1、这是跨平台的拱趴大...
有挂透视!竞技联盟透视插件,微... 您好,竞技联盟透视插件这款游戏可以开挂的,确实是有挂的,需要了解加去威信【485275054】很多玩...
受玩家影响!德州之星扫描器,斗... 受玩家影响!德州之星扫描器,斗棋有bug么开挂,手段教程(讲解有挂)-哔哩哔哩1、下载好德州之星扫描...
近日!透视辅助器免费蜀山四川,... 近日!透视辅助器免费蜀山四川,手机填大坑辅助器,法子教程(有挂解惑)-哔哩哔哩1、打开软件启动之后找...
有挂透视!hhpkoer辅助器... 有挂透视!hhpkoer辅助器视频,微乐小程序黑科技下载,方针教程(有挂方略)-哔哩哔哩1、首先打开...