zoukankan      html  css  js  c++  java
  • 3090显卡(CUDA11.1)安装Pytorch

    由于官方暂时没有11.1版本的Pytorch安装方法,可以使用11.0版本的安装方法,经过测试,在3090上是可以正常使用的。
    官方给出的安装方法:
    pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

    在安装过程中,由于连接的是Pytorch的官方地址,所以下载的速度是很慢的,可以手动下载torch包和torchvison的包进行安装,运行一下上面的安装命令,就能看到地址。
    另外,安装完torch包以后,在安装torchvison的包时,可能会自动下载另一个版本的torch包,可以通过
    pip install --no-deps torchvision-0.8.2+cu110-cp38-cp38-linux_x86_64.whl解决。

  • 相关阅读:
    min_25筛入门
    [湖南集训]更为厉害/谈笑风生
    [ARC060D] 最良表現
    [CQOI2007]矩形
    [SCOI2009]粉刷匠
    PAT乙级1030
    PAT乙级1028
    PAT乙级1029
    PAT乙级1026
    PAT乙级1027
  • 原文地址:https://www.cnblogs.com/shubin/p/14141316.html
Copyright © 2011-2022 走看看