1.查自己电脑的cuda版本

win+R 打开运行,cmd 使用命令 nvcc --version

有GPU装pytorch

 

2.在官网上找到合适的命令
有GPU装pytorch

3.在prompt中安装
可能我网好吧,一下子就装好了

有GPU装pytorch

4.检查是否安装成功

有GPU装pytorch

相关文章: