安装GPU版本Pytorch（全网最详细过程）

目录

一、前言

二、安装CUDA

三、安装cuDNN

四、安装Anacanda

五、安装pytorch

六、总结

一、前言

最近因为需要安装GPU版本的Pytorch，所以自己在安装过程中也是想着写一篇博客，把整个过程记录下来，在整个过程中也遇到了不少的问题，查看了以往别人的一些解决方案，希望能够给遇到同样问题的提供一些参考。

如果大家在安装pytorch的时候，并不需要安装gpu版本的，即使用cpu进行训练的pytorch，那么直接在Aanaconda Prompt当中输入以下命令即可：

pip install pytorch

如果是想要安装gpu版本的pytorch，则需要参考以下几个步骤。

二、安装CUDA

以我自己的电脑为例：我的电脑显卡是Geforce MX150，CPU是i5-8250u，操作系统是win10。如果想要使用gpu版本的pytorch，首先得安装一个自己显卡支持的CUDA版本。

根据我自己的电脑，选择的是CUDA9.2版本。

1、查看电脑的显卡驱动版本：右键选择NVIDIA控制面板→帮助→系统信息

2、查看安装不同版本CUDA所对应的显卡驱动版本（比如这里我选择的就是9.2版本，建议显卡驱动版本高的安装高版本CUDA，显卡驱动版本只要大于CUDA对应的最低版本就行，显卡驱动版本版本过低可以升级）：[https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html](https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html "https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html")

 3、接下来下载所选择版本的CUDA，下载网址：[https://developer.nvidia.com/cuda-toolkit-archive](https://developer.nvidia.com/cuda-toolkit-archive "https://developer.nvidia.com/cuda-toolkit-archive")

4、下载完成直接安装即可。