win10安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录

2022-02-26

Word count: 707 | Reading time≈ 2 min

最近由于机器学习大作业需要用到神经网络，记录一下配置环境的过程。
前人铺路，我只是结合自己的环境做了一个简单的总结，写的不好多多见谅。

1.安装python3.9

首先是python的安装，选用当前时间最新版的python3.9.1
官网下载安装包即可https://www.python.org/downloads/

一路“下一步”，建议安装到默认路径。

然后，配置环境变量：
找到安装的位置，把图中所示的两个路径加入环境变量。
我的路径是：
C:\Users\Administrator\AppData\Local\Programs\Python\Python39
C:\Users\Administrator\AppData\Local\Programs\Python\Python39\Scripts

补充一点：
conda里python中的pytorch比较老，遇到没法调用的问题（这里没有深究），所以没有用anaconda，而是直接官网找最新的python。

2.安装cuda（需要vs环境，我的是vs2015）

CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。

我当前的环境是vs2015，电脑配置是

注意CUDA的版本不能超过GPU的版本。

官网的CUDA安装路径如下：
https://developer.nvidia.com/zh-cn/cuda-downloads

圈出来的部分，前者是一个比较大的安装包，下载到本地安装，后者是很小的安装包，但是需要联网，我选的前者。

仍然选择当前最新的版本11.1，安装到默认路径下：
C:\Program Files\NVIDIA GPU Computing Toolkit

3.安装cudnn（版本和cuda对应，更老的版本也可以但不建议）

NVIDIA cuDNN是用于深度神经网络的GPU加速库。
老规矩，走官网。
https://developer.nvidia.com/rdp/cudnn-archive

这里的下载需要注册，有个技巧是复制链接，在迅雷打开可以直接下载。

下载好的压缩包解压之后，会发现三个文件夹。
分别把文件放到cuda路径对应的文件夹下。

4.安装pytorch

它是一个基于Python的可续计算包，提供两个高级功能：1、具有强大的GPU加速的张量计算（如NumPy）。2、包含自动求导系统的深度神经网络。
官网：
https://pytorch.org/get-started/locally/

可以通过pip安装，最下面的命令行在cmd中打开即可。
由于速度慢，所以直接从别人那里拷贝过来了。