分析原因:显卡算力需要与torch算力匹配
安装较高版本的torch,cuda11以上。目前笔者安装的版本如下:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
分析原因:显卡算力需要与torch算力匹配
安装较高版本的torch,cuda11以上。目前笔者安装的版本如下:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
2万+

被折叠的 条评论
为什么被折叠?