一开始安装torch的时候没有指定cuda版本,所以安装的是cpu版本,无法调用gpu
可以使用如下安装方法:
pip install torch -f https://download.pytorch.org/whl/cu118/torch_stable.html(应该可以用)
pip install torch --index-url https://download.pytorch.org/whl/nightly/cu118(亲测有效,但是nightly表示PyTorch 开发人员每天编译和生成的版本，包含了最新的代码和功能,不稳定)