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

相关内容

热门资讯

第八分钟了解!都莱辅助软件(辅... 第八分钟了解!都莱辅助软件(辅助挂)详细透视开挂辅助wpk教程(好像是有挂)-哔哩哔哩相信很多朋友都...
总算清楚!"德州免费... 总算清楚!"德州免费辅助神器app"详细外挂透明挂辅助神器-先前是真的有挂(哔哩哔哩);1、总算清楚...
3分钟了解!蜀山四川麻将可以让... 3分钟了解!蜀山四川麻将可以让系统发好牌(辅助挂)原来是真的有挂(专业辅助黑科技教程)-哔哩哔哩;1...
辅助透视!werplan可以透... 辅助透视!werplan可以透视挂(透视)底牌透视挂辅助方法(可靠开挂辅助解密教程)-哔哩哔哩;we...
四分钟了解!浙江宝宝游戏辅助器... 四分钟了解!浙江宝宝游戏辅助器(辅助挂)详细透视开挂辅助2025新版技巧(果然存在有挂)-哔哩哔哩;...
安装程序教程"微扑克... 您好:微扑克内置这款游戏可以开挂的,确实是有挂的,很多玩家在这款游戏中打牌都会发现很多用户的牌特别好...
透视代打!拱趴大菠萝可以开挂(... 透视代打!拱趴大菠萝可以开挂(透视)底牌透视挂辅助软件(可靠开挂辅助解密教程)-哔哩哔哩;拱趴大菠萝...
第八分钟了解!情怀麻将(辅助挂... 第八分钟了解!情怀麻将(辅助挂)原先是真的有挂(专业辅助黑科技教程)-哔哩哔哩;情怀麻将是一项情怀麻...
传递经验"微扑克模拟... 传递经验"微扑克模拟器"详细外挂透明挂辅助下载-往昔存在有挂(哔哩哔哩);1、让任何用户在无需AI插...
第8分钟了解!约战竞技场辅助器... 【福星临门,好运相随】;第8分钟了解!约战竞技场辅助器(辅助挂)详细透视开挂辅助解密教程(果然真的有...