安装GPU版本的torch,利用pytorch官网的pip和conda命令下载的不是GPU版本的torch,torch后面没有+cu118.
自己使用国内镜像源安装指定版本的torch,torchvision,torchaudio
下面展示一些 内联代码片
。
pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package --trusted-host mirrors.aliyun.com
注意torch版本和torchvision版本一定要匹配,torchaudio和torch一样就可以。cu版本也要统一。