笔记本1050ti跑autoformer模型,环境搭建过程

## 1、选显卡对应得驱动程序

https://www.nvidia.com/Download/index.aspx
notebook是笔记本,下载类型选sd。
不更新驱动会报:RuntimeError: The NVIDIA driver on your system is too old (found version 8000).
Please update your GPU driver by downloading and installing a new version from the URL:
http://www.nvidia.com/Download/index.aspx
必须下载官网得驱动程序更新,自己在设备里更新驱动没用,设备里更新还会导致第二步得cuda安装失败。

## 2、根据显卡的驱动版本,找对应的cuda版本

只要在第一步更新了驱动,那就能支持最新的cuda。
但是,太新的cuda装了报错,随便选个老点的版本不一定行。应该要用的根据pytorch版本,决定cuda版本。
https://developer.nvidia.com/cuda-toolkit-archive
22年的论文很多用的pytorch1.9.0,我用这个版本试了cuda 12,11,10等五六个cuda版本,下面这个链接也说python3.6版本可以装pytorch1.9,用cuda11.3,但试了发现识别不了gpu,**最后发现只有cuda 11.3的基础上,装python3.7,再装pytorch1.10.1,才能识别出gpu**。
https://blog.csdn.net/qq_43599739/article/details/123015331

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
而https://pytorch.org/官方推荐的pytorch最新版本,让用cuda11.8或12.1.
如果是**新电脑,就装cuda 12.1,再装最新pytorch**。

posted @ 2024-04-27 11:27  zhaot1993  阅读(37)  评论(0编辑  收藏  举报