Windows的python配置pytorch、torchvision、NVIDIA(英伟达)、cuDNN进行gpu加速计算
相信很多小伙伴在跑深度学习神经网络的时候用cpu跑又慢又占内存,本文将介绍如何将自己电脑增加gpu跑模型的路线,方便加速计算。废话不多说,下面咱们直接开始:一、python配置建议下载的python版本为3.5到3.10之间,因为大多数用于gpu加速计算的torch都在这个版本区间。一般常用的是python3.6或者3.8。下面我们就以python3.8为基础环境进行配置,其他版本下面操作类似。如