pytorch 1.0对应的torchvision

最新推荐文章于 2024-07-09 12:43:02 发布

m0_61107925

最新推荐文章于 2024-07-09 12:43:02 发布

阅读量907

点赞数

分类专栏： Environments 文章标签： pip

本文链接：https://blog.youkuaiyun.com/m0_61107925/article/details/119871604

版权

Environments 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了在遇到PyTorch与旧版CUDA冲突时如何通过降低PyTorch版本解决问题，同时提供了清华镜像源地址。重点在于适配旧驱动环境下的深度学习库使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pytorch 1.0对应的torchvision版本为0.2

torchvision 0.3对应的版本为torch 1.5

code running error:

AssertionError:
The NVIDIA driver on your system is too old (found version 10000).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.

解决：

降低pytorch的版本即可;

pip清华镜像源地址：

https://pypi.tuna.tsinghua.edu.cn/simple