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

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

1.安装python3.9

首先是python的安装,选用当前时间最新版的python3.9.1
官网下载安装包即可https://www.python.org/downloads/
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第1张图片

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

然后,配置环境变量:
找到安装的位置,把图中所示的两个路径加入环境变量。
我的路径是:
C:\Users\Administrator\AppData\Local\Programs\Python\Python39
C:\Users\Administrator\AppData\Local\Programs\Python\Python39\Scripts
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第2张图片
补充一点:
conda里python中的pytorch比较老,遇到没法调用的问题(这里没有深究),所以没有用anaconda,而是直接官网找最新的python。

2.安装cuda(需要vs环境,我的是vs2015)

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。

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

win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第3张图片
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第4张图片
注意CUDA的版本不能超过GPU的版本。

官网的CUDA安装路径如下:
https://developer.nvidia.com/zh-cn/cuda-downloads
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第5张图片
圈出来的部分,前者是一个比较大的安装包,下载到本地安装,后者是很小的安装包,但是需要联网,我选的前者。
在这里插入图片描述

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

3.安装cudnn(版本和cuda对应,更老的版本也可以但不建议)

NVIDIA cuDNN是用于深度神经网络的GPU加速库。
老规矩,走官网。
https://developer.nvidia.com/rdp/cudnn-archive
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第6张图片

这里的下载需要注册,有个技巧是复制链接,在迅雷打开可以直接下载。
在这里插入图片描述

下载好的压缩包解压之后,会发现三个文件夹。
分别把文件放到cuda路径对应的文件夹下。
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第7张图片

4.安装pytorch

它是一个基于Python的可续计算包,提供两个高级功能:1、具有强大的GPU加速的张量计算(如NumPy)。2、包含自动求导系统的深度神经网络。
官网:
https://pytorch.org/get-started/locally/
win10 安装python3.9.1+cuda11.1+cudnn+pytorch+opencv记录_第8张图片
可以通过pip安装,最下面的命令行在cmd中打开即可。
由于速度慢,所以直接从别人那里拷贝过来了。

可以在python中通过下面的代码来检验,True表示pytorch的cuda配置成功
在这里插入图片描述

5.安装opencv

方法一:
打开cmd,输入pip install opencv-python即可,但超级慢还可能断开连接。
此处建议清华镜像下载

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

方法二:
下载相应Python版本的OpenCV的whl文件
https://www.lfd.uci.edu/~gohlke/pythonlibs/#opencv
然后打开cmd,在whl文件对应文件夹路径下,使用pip安装,如
在这里插入图片描述

至此我需要的环境已经配置完成。

你可能感兴趣的:(神经网络,python)