Windows 下使用Pytorch-GPU

Windows 下使用Pytorch-GPU

1. 检查、更新NVIDIA显卡驱动版本

下载https://www.nvidia.cn/geforce/geforce-experience/更新驱动

2. 查看CUDA版本（支持的最高版本）

https://www.jianshu.com/p/d3b9419a0f89

3. 根据CUDA最高版本、需要使用的Pytorch版本，下载安装CUDA与Pytorch

原理详见：https://blog.csdn.net/kellyroslyn/article/details/109668001

上面网页中，对应关系的内容已经比较陈旧。具体对应关系，请看：https://www.cnblogs.com/tingtin/p/13601104.html

• 选择CUDA相应版本安装：

  • https://developer.nvidia.com/cuda-toolkit-archive

• 选择Pytorch相应版本，在命令行输入对应的命令安装：

  • https://pytorch.org/get-started/previous-versions/

  每条安装命令都注明了Pytorch版本、系统、支持的CUDA版本

当然，如果配环境让你十分头疼、或者你的GPU性能并没有那么优秀。那么，忘记这些东西吧，简简单单地用CPU随便跑跑，感受深度学习的魅力吧。