动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习

本文为自己安装过程记录,具体教程可查看:李沐老师教学视频


目录

    • 1. 查看电脑的GPU信息,确认有Nvidia GPU
    • 2.安装CUDA
    • 3.安装miniconda
    • 4.pytorch官网下载GPU版本
    • 5.《动手学深度学习》下载Jupyter记事本
    • 6.安装d2l和Jupyter
    • 待解决问题


1. 查看电脑的GPU信息,确认有Nvidia GPU

动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第1张图片
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第2张图片

另:也可以直接通过任务管理器查看
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第3张图片

2.安装CUDA

下载地址:https://developer.nvidia.com/cuda-downloads
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第4张图片
在命令行键入如下命令查看信息:

nvidia-smi

动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第5张图片

3.安装miniconda

下载地址:https://docs.conda.io/en/latest/miniconda.html
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第6张图片
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第7张图片
查看版本信息:
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第8张图片

4.pytorch官网下载GPU版本

下载地址:https://pytorch.org/get-started/locally/
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第9张图片
直接复制下载命令执行:
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第10张图片
这儿有warning,但后面运行好像没啥影响,验证一下:
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第11张图片

5.《动手学深度学习》下载Jupyter记事本

网址:https://zh-v2.d2l.ai/
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第12张图片

6.安装d2l和Jupyter

pip install jupyter d2l

安装完成继续键入jupyter notebook就可以运行Jupyter实例啦~


待解决问题

动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第13张图片
运行报错:
RuntimeError: DataLoader worker (pid(s) 23356, 21548, 14108, 8572) exited unexpectedly
动手学深度学习——Windows下安装CUDA和Pytorch跑深度学习_第14张图片
应该是内存空间不足,目前还没跑出来,先继续学着后面解决了再更。

你可能感兴趣的:(pytorch,深度学习,python)