【2021最新】Windows10安装GPU版pytorch

之前的很多教程要自己下载安装cudatoolkit, cudnn等,非常麻烦。现在pytorch已经支持在cmd一条命令下载安装了。

  1. 首先需要安装anaconda,已经有的同学可以忽略这一步
    没有anaconda的请先下载安装:https://www.anaconda.com/products/individual

  2. 查看自己的Cuda版本

nvcc-V

【2021最新】Windows10安装GPU版pytorch_第1张图片
这里我的版本是11.1

  1. 在pytorch官网上找到匹配的版本安装

https://pytorch.org/
【2021最新】Windows10安装GPU版pytorch_第2张图片
这里没有的话看之前的版本就好。
接下来是漫长的安装过程~

你可能感兴趣的:(pytorch,windows,anaconda)