可能是下成了cpu版本的torch,可以查看下
print(torch.__version__)
解决方法:
直接下载三件套:
打开pycharm终端,输入pip install torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116,回车
最终效果:
可能是下成了cpu版本的torch,可以查看下
print(torch.__version__)
解决方法:
直接下载三件套:
打开pycharm终端,输入pip install torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116,回车
最终效果: