cuda_pytorch_12.8_安装

最新推荐文章于 2025-06-09 21:10:21 发布

李乐童

最新推荐文章于 2025-06-09 21:10:21 发布

阅读量932

点赞数 3

CC 4.0 BY-SA版权

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/weixin_54467758/article/details/148188475

显卡版本： NVIDIA GeForce RTX 5070Ti

卸载之前安装的CPU版本

 pip uninstall torch torchvision torchaudio

安装pytorch-GPU版本(12.8版本)
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple --index-url https://download.pytorch.org/whl/cu128

pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple --index-url https://download.pytorch.org/whl/cu128