1、确认具有NVIDIA GPU
方案1:win+r,输入dxdiag,弹出窗口,切换到显示选项栏,即可看到显卡信息
方案2:任务管理器 → 性能 → 下拉可看到GPU,点击查看详情
2、安装CUDA
查看系统支持的CUDA版本
控制面板 →所有控制面板项 → NVIDIA控制面板
帮助 → 系统信息 → 组件 → NVCUDA64这一项有版本,例如11.2
官网下载CUDA
谷歌/百度搜索 download cuda (+版本号例如11.2)
Windows → x86_64 → 10 →exe(local)→ download
安装CUDA
下载完后打开,可一路默认安装(在c盘!)也可自定义安装
3、安装miniconda
https://docs.conda.io/en/latest/miniconda.html
官网下载,Windows installers下
例如Python3.8 Miniconda3 Windows 64-bit
下载完后打开,可一路默认安装(在c盘!)也可自定义安装
安装完miniconda后,桌面左下角Windows,找到Anaconda一栏中打开PowerShell
python --version
可查看python版本
4、安装GPU版 Pytorch
https://pytorch.org/ 官网下载pytorch
GetStarted → Stable(1.9.1) → Windows → Pip → python → Cuda 11.1
下方会生成一条命令 复制到Anaconda的powershell运行安装
5、安装d2l和Jupyter
pip install jupyter d2l
6、下载d2l记事本运行测试
在d2l课程主页下载代码文件运行
https://zh-v2.d2l.ai/
在powershell下进入代码文件夹下 输入jupyter notebook
系统自动弹出页面即可查看运行