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