链接:https://download.pytorch.org/whl/torch_stable.html
下载pytorch-torchvision-torchaudio等一系列一定要版本匹配,并且如果是在gpu上跑的话,一定要都是cu版本
参考链接:https://blog.csdn.net/AiTanXiing/article/details/129933911
然后先下载上面的离线版本,版本一定要匹配,去官网看https://pytorch.org/get-started/previous-versions/
然后我这里成功配对的是这三个:
然后下载离线的是这三个:
对应的python3.9,nvcc -V出来的是cuda12.0,只能向下兼容,所以对应的是11.8版本。
安装代码如下:
pip3 install torch-2.0.1+cu118-cp39-cp39-linux_x86_64.whl -i http://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com
pip3 install torchvision-0.15.2+cu118-cp39-cp39-linux_x86_64.whl -i http://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com
pip3 install torchaudio-2.0.2+cu118-cp39-cp39-linux_x86_64.whl -i http://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com
最后出来的视频效果比较差,很卡顿,并且识别效果也较差。