原因:直接使用pip install torch是默认安装的cpu版本,我们需要到pytorch的指定官网去下载安装torch
我们首先在环境里面卸载掉
pip uninstall torch
在环境中使用nvidia查看自己的cuda版本之后
到pytorch官网进行下载安装。(需要注意的是,下载的cuda版本不能超过自己电脑上那个硬件的CUDA版本,比如上面查到本机的CUDA版本是11.7,那么只能安装11.7以下的CUDA版本)。
在环境中执行以下命令即可
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116