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

07-19 1782阅读

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

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

    这里输出的也是False

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

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

    解决方法

    应该将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

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

    这里输出是True

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

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

VPS购买请点击我

文章版权声明:除非注明,否则均为主机测评原创文章,转载或复制请以超链接形式并注明出处。

目录[+]