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