安装pytorch的gpu版本时,误装成了cpu版本

原因:直接使用pip install torch是默认安装的cpu版本,我们需要到pytorch的指定官网去下载安装torch

我们首先在环境里面卸载掉

pip uninstall torch

在环境中使用nvidia查看自己的cuda版本之后

安装pytorch的gpu版本时,误装成了cpu版本_第1张图片到pytorch官网进行下载安装。

安装pytorch的gpu版本时,误装成了cpu版本_第2张图片 

在环境中执行以下命令即可

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

 

你可能感兴趣的:(Python入门知识,pytorch,深度学习,python)