windows10安装GPU版pytorch步骤（全程离线）

灰太狼 2023-07-08 02:49 13阅读 0赞

# 1、首先安装cuda10.0和cudnn。 #

软件：  
cuda\_10.0.130\_411.31\_win10  
cudnn-10.0-windows10-x64-v7.6.4.38  
下载的cuda包括显卡驱动，所以不用单独安装显卡，再安装cuda。  
参考：  
[https://blog.csdn.net/qq\_37296487/article/details/83028394][https_blog.csdn.net_qq_37296487_article_details_83028394]

# 2、离线安装pytorch #

## 2.1 下载软件 ##

torch-1.2.0-cp36-cp36m-win\_amd64.whl  
torchvision-0.4.0-cp36-cp36m-win\_amd64.whl

进入[https://download.pytorch.org/whl/torch\_stable.html][https_download.pytorch.org_whl_torch_stable.html]根据cuda版本和python版本下载相对应的pytorch版本。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2Vyc29uZw_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2Vyc29uZw_size_16_color_FFFFFF_t_70 1]

## 2.2 安装步骤 ##

D:\cuda文件>pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
    Processing d:\cuda文件\torch-1.2.0-cp36-cp36m-win_amd64.whl
    Requirement already satisfied: numpy in d:\anaconda3\lib\site-packages (from torch==1.2.0) (1.14.3)
    distributed 1.21.8 requires msgpack, which is not installed.
    Installing collected packages: torch
    Successfully installed torch-1.2.0

有时安装成功，但是会报缺少msgpack包，下载[msgpack-python和msgpack包][msgpack-python_msgpack]，并离线安装。

D:\cuda文件>pip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
    Processing d:\cuda文件\torchvision-0.4.0-cp36-cp36m-win_amd64.whl
    Requirement already satisfied: six in f:\programdata\anaconda3\lib\site-packages (from torchvision==0.4.0) (1.11.0)
    Requirement already satisfied: numpy in f:\programdata\anaconda3\lib\site-packages (from torchvision==0.4.0) (1.14.3)
    Requirement already satisfied: pillow>=4.1.1 in f:\programdata\anaconda3\lib\site-packages (from torchvision==0.4.0) (5.1.0)
    Requirement already satisfied: torch in f:\programdata\anaconda3\lib\site-packages (from torchvision==0.4.0) (1.2.0)
    Installing collected packages: torchvision
    Successfully installed torchvision-0.4.0

安装成功后查看torch版本。

C:\Users\XXX>python
    Python 3.6.5 |Anaconda, Inc.| (default, Mar 29 2018, 13:32:41) [MSC v.1900 64 bit (AMD64)] on win32
    Type "help", "copyright", "credits" or "license" for more information.
    >>>
    >>> import torch
    >>>
    >>> print(torch.__version__)
    1.2.0
    >>>

[https_blog.csdn.net_qq_37296487_article_details_83028394]: https://blog.csdn.net/qq_37296487/article/details/83028394
[https_download.pytorch.org_whl_torch_stable.html]: https://download.pytorch.org/whl/torch_stable.html
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2Vyc29uZw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200224212703141.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2Vyc29uZw==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2Vyc29uZw_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200224212817156.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb2Vyc29uZw==,size_16,color_FFFFFF,t_70
[msgpack-python_msgpack]: https://download.csdn.net/download/maoersong/12188997