使用requirements.txt文件安装cuda(GPU)版本的pytorch

使用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

相关推荐
凪卄1213几秒前
图像预处理 二
人工智能·python·深度学习·计算机视觉·pycharm
巫婆理发2223 分钟前
强化学习(第三课第三周)
python·机器学习·深度神经网络
AI赋能18 分钟前
自动驾驶训练-tub详解
人工智能·深度学习·自动驾驶
seasonsyy19 分钟前
1.安装anaconda详细步骤(含安装截图)
python·深度学习·环境配置
deephub26 分钟前
AI代理性能提升实战:LangChain+LangGraph内存管理与上下文优化完整指南
人工智能·深度学习·神经网络·langchain·大语言模型·rag
EulerBlind28 分钟前
【运维】SGLang 安装指南
运维·人工智能·语言模型
心之语歌30 分钟前
Spring AI MCP 客户端
人工智能·spring·github
半新半旧37 分钟前
python 整合使用 Redis
redis·python·bootstrap
go54631584651 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
Blossom.1181 小时前
基于深度学习的图像分类:使用Capsule Networks实现高效分类
人工智能·python·深度学习·神经网络·机器学习·分类·数据挖掘