1. .to(device)卡住
可能是pytorch版本太老了,安装新的试一下。(有一些答案说,使用conda安装的缘故,换成pip安装)
- 实际操作,创建新的虚拟环境,安装python3.9版本,使用命令安装:
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3
- conda默认使用清华源
2. 使用conda install pytorch torchvision cudatoolkit=xxx安装,pytorch是cpu版本
没有对应版本的gpu版本,在安装前需要检查是否是gpu版本,参考链接