美文网首页
Pytorch安装最稳超详细——(需VPN)

Pytorch安装最稳超详细——(需VPN)

作者: hdjwuxvks | 来源:发表于2023-07-30 13:42 被阅读0次

    ## 安装GeForce驱动程序

    ### 下载

    [英伟达官网](https://www.nvidia.cn/geforce/drivers/)

    手动搜索驱动程序:

    产品类型:GeForce

    产品系列:是什么选什么,没有就找最接近的(后面还有产品进一步找你的显卡),Notebooks表示笔记本,一般选这个。

    产品、操作系统、语言选自己的

    下载类型:选Studio就好,没有就选全部或者Game Ready都可以

    最后开始搜索

    搜索结果一般可以随便选,选最新的或者比较新的都可以。

    ### 安装

    注意!安装之前关闭火绒!

    用管理员身份运行

    可以自定义,安装图形驱动程序就好,PhysX系统软件一般不用安,Experience不用(安也没事),可以执行清洁安装(反正也没什么配置不是,又不玩游戏)

    等待安装成功

    ## 安装CUDA Toolkit

    打开cmd(管理员最好),输入

    ```

    nvidia-smi

    ```

    出来个框框就说明上一步驱动程序安装成功。

    ### 下载

    在框框的右上角有个“CUDA Version:”,记作【CUDA Version】比如12.2,就是你能安装的CUDA Toolkit的最高版本,可以就安那个版本,也可以稍低一点。不用担心和你的pytorch所需的CUDA不同,可以向下兼容的,你下pytorch的时候下所需版本完全没问题;反而,如果你刚好安装你所需版本的CUDA Toolkit,反而可能会有问题,我就有这样的经历。

    在下面这个网址下载,把12-2改成你的【CUDA Version】就好,注意是-不是.

    ```

    https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

    ```

    选择你的配置(Windows,10,exe(local))

    点击Download就好,很快的。

    ### 安装

    用管理员身份运行

    安装同上,Experience也可以不安

    等待安装成功

    打开cmd(管理员最好),输入

    ```

    nvcc -V

    ```

    能看到版本之类的信息。

    ## 安装Pytorch

    进入[网址](https://pytorch.org/get-started/locally/)

    红框框那,选择Stable,Windows,pip(注意要点pip),python,如果有你所需的CUDA版本就点他,没有就选稍高些的版本,或者点击这个[链接](https://pytorch.org/get-started/previous-versions/)进入之前版本下载(Ctrl F 查找,比如版本11.6就搜116,注意要是pip)。

    找到链接复制下来

    ```

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117(框框那)

    或者

    pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116(之前版本那)

    ```

    进入要安装Pytorch的虚拟环境,在上面的链接后面加你的系统代理,"--proxy http://127.0.0.1:1080" ,特别注意系统代理的端口,(用pip安装其他包也可以这么做)

    ```

    pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 --proxy http://127.0.0.1:1080

    ```

    ### 验证

    ```

    python

    import torch

    torch.cuda.is_available()

    ```

    返回True,说明成功

    ***

    修曦所著,转载注明

    如果能帮到人就更好啦

    相关文章

      网友评论

          本文标题:Pytorch安装最稳超详细——(需VPN)

          本文链接:https://www.haomeiwen.com/subject/kxdspdtx.html