一:去官网找对应版本——https://pytorch.org/

NLP:正确安装Pytorch

二,找对应版本

NLP:正确安装Pytorch

如果不知道CUDA版本参考如下:

控制面板——小图标查看——NVIDIA控制面板——帮助——系统信息——组件——3D设置

NLP:正确安装Pytorch

三,直接运行官方给的命令即可

NLP:正确安装Pytorch

 

相关文章: