Pytorch1.4 GPU环境

发布 : 2020-04-05 分类 : 杂记 浏览 :

Pytorch1.4 GPU环境

首先确定你的电脑显卡是NVIDIA的,如果不是,请安装CPU版本。使用pip直接安装:

1
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

驱动

1586089962427.png

  • 双击安装:
    1586090360189.png
    1586090532529.png

    显卡驱动按照自己情况安装:
    1586090713816.png

    根据自己要求,选择安装位置
    1586090591332.png

    CUDNN

    下载与CUDA10.1匹配的CUDNN。需要登录NVIDIA开发者社区
    1586091216025.png

  • 解压后复制到CUDA Development目录

1586091415721.png

环境变量

查看环境变量是否自动添加了,如果没有就自己添加一下bin目录
1586091834271.png

测试

查看是否安装成功
1586091899039.png

安装Pytorch

安装好anaconda后,用pip工具安装pytorch:

1
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html


安装好后,测试:
1586092015000.png
如果返回True则Pytorch GPU版安装完成。

本文作者 : HeoLis
原文链接 : http://ishero.net/Pytorch1.4%20GPU%E7%8E%AF%E5%A2%83.html
版权声明 : 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明出处!

学习、记录、分享、获得

微信扫一扫, 向我投食

微信扫一扫, 向我投食