使用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、这是跨平台的...
黑科技功能(wePokE)外挂... 黑科技功能(wePokE)外挂透明挂黑科技辅助插件(透视)2025新版教程(2025已更新)(哔哩哔...
一分钟了解!wepower提高... 一分钟了解!wepower提高中牌率,微信决胜麻将助赢神器,德州论坛(真是有挂)-哔哩哔哩;亲,关键...
黑科技透明挂(AaPOKER)... 黑科技透明挂(AaPOKER)外挂透明挂黑科技辅助软件(透视)透明教程(2021已更新)(哔哩哔哩)...
9分钟了解!wepoke软件规... 9分钟了解!wepoke软件规律,八一字牌辅助软件,解密教程(果真有挂)-哔哩哔哩1)八一字牌辅助软...
黑科技智能(WEPoke)外挂... 黑科技智能(WEPoke)外挂透明挂黑科技辅助插件(透视)德州论坛(2021已更新)(哔哩哔哩)1、...
一分钟了解!微扑克ai分析,牵... 一分钟了解!微扑克ai分析,牵手字牌有挂吗,插件教程(讲解有挂)-哔哩哔哩一分钟了解!微扑克ai分析...
黑科技最新(Wpk)外挂透明挂... 黑科技最新(Wpk)外挂透明挂黑科技辅助代打(透视)2025版教程(2026已更新)(哔哩哔哩)1、...
第3分钟了解!wpk智能ai,... 第3分钟了解!wpk智能ai,博雅红河棋牌如何拿到好牌,攻略教程(竟然有挂)-哔哩哔哩1.博雅红河棋...
黑科技设备(德扑平台)外挂透明... 黑科技设备(德扑平台)外挂透明挂黑科技辅助软件(透视)攻略教程(2025已更新)(哔哩哔哩);一、德...