pip安装cuda版torch

一开始安装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 开发人员每天编译和生成的版本，包含了最新的代码和功能,不稳定)

官方安装方式：https://pytorch.org/get-started/previous-versions/