安装cpu版本,使用如下命令:
pip install torch1.7.0+cpu torchvision0.8.1+cpu torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
安装Gpu版本,使用如下命令:
pip install torch=1.7.0 torchvision=0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)