之前的很多教程要自己下载安装cudatoolkit, cudnn等,非常麻烦。现在pytorch已经支持在cmd一条命令下载安装了。
-
首先需要安装anaconda,已经有的同学可以忽略这一步
没有anaconda的请先下载安装:https://www.anaconda.com/products/individual -
查看自己的Cuda版本
nvcc-V
这里我的版本是11.1
- 在pytorch官网上找到匹配的版本安装
https://pytorch.org/
这里没有的话看之前的版本就好。
接下来是漫长的安装过程~