深度学习GPU环境搭建:
安装步骤:
1、安装GPU驱动:官网有(可以理解为告诉电脑这个东西怎么使用)
不同厂家的不一样,用于深度学习的一般是NVIDIA、AMD家。
本人是NVIDIA家,驱动长这样子:
2、python环境里配置cuda环境:这里是让程序能够用GPU跑代码,不然只能用CPU跑代码。
需要安装两个软件:
(1)CUDA toolkit Download
(2)cuDNN Download
安装步骤参见:
https://blog.csdn.net/weixin_43848614/article/details/117221384