Windows下安装配置深度学习环境
1. 准备工作
1.1 环境准备
- 操作系统:win10 22H2
- GPU:Nvidia GeForce RTX 3060 12G
1.2 安装Nvidia驱动、cuda、cuDNN
下载驱动需要注册并登录英伟达账号。我这里将下面用到的安装包放到了百度网盘,可以关注微信公众号思享云栈回复deeplearning获取。
到NVIDIA驱动程序下载并安装适合你显卡的最新驱动,我这里下载的安装包版本为555.85-desktop-win10-win11-64bit-international-dch-whql.exe
,双击进行安装即可。
Windows下安装后路径为:C:\Program Files\NVIDIA Corporation
1.3 安装CUDA Toolkit
官网查看CUDA版本与驱动版本的配套关系,部分截取如下:
到CUDA下载界面下载CUDA安装程序。我这里下载安装CUDA Toolkit 11.8
,满足配套关系,下载后的安装包为cuda_11.8.0_522.06_windows.exe
,傻瓜式安装即可。
Windows下安装后路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
Windows下会自动添加环境变量配置:
CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
打开cmd命令行&#x