1. 环境准备与版本选型别急着敲命令先把账算清楚装 CUDA 这件事十个人里有八个是栽在“版本对不上”这五个字上。我见过太多人显卡插上去驱动随便找个最新版一装CUDA Toolkit 也挑最新的下载结果跑起 PyTorch 或者 TensorFlow 直接报CUDA error: no kernel image is available for execution on the device然后开始怀疑人生。问题不在你手笨在于 NVIDIA 这套东西的版本依赖链是驱动版本 → CUDA Runtime 版本 → 框架编译版本三层锁死的任何一层错位都会炸。先把最核心的对应关系讲透。N 卡驱动Driver是操作系统层面的它决定了你这台机器最高能支持到哪个 CUDA 版本。注意是“最高”不是“必须”。比如驱动版本 550.144.03它支持 CUDA 12.4 及以下的所有版本。你装 CUDA 11.8 完全没问题装 CUDA 12.4 也行但你想装 CUDA 12.5 就得先升驱动。这个“向下兼容”的特性是很多人没搞明白的以为驱动和 CUDA 必须一一对应其实不是。那怎么查驱动支持的最高 CUDA 版本最直接的办法是看nvidia-smi右上角那行CUDA Version: 12.4。这里有个坑要提前说nvidia-smi显示的 CUDA Version 是驱动支持的最高版本不是你当前安装的 CUDA Toolkit 版本。很多人看到这里显示 12.4就以为自己已经装了 CUDA 12.4然后去nvcc -V一看发现根本没有或者版本对不上就开始懵。这两个东西是分开的nvidia-smi属于驱动层nvcc属于 Toolkit 层后面我会详细拆。选版本的正确姿势是这样的先确定你要跑的框架需要什么 CUDA 版本再反推驱动版本。举个例子TensorFlow 2.5.0 官方要求 CUDA 11.2 cuDNN 8.1那你就得装 CUDA 11.2驱动只要满足最低要求 450.80.02就行。再比如 PyTorch 2.x 现在主流是 CUDA 11.8 和 12.1 两个版本你选哪个取决于你的显卡架构和驱动。4060 Ti 是 Ada Lovelace 架构算力 8.9CUDA 11.8 和 12.x 都支持但如果你要用最新的 FlashAttention 或者某些编译型库12.x 会更省心。这里插一句关于wsl2安装cuda的事。WSL2 里的 CUDA 安装逻辑和纯 Linux 不太一样。WSL2 本身不需要你在里面装显卡驱动驱动是装在 Windows 宿主机的WSL2 通过/usr/lib/wsl/lib这个路径直接调用宿主机的驱动。所以你只需要在 WSL2 里装 CUDA Toolkit 就行千万别在 WSL2 里再装一遍 Linux 驱动装了反而会冲突。这个点后面会展开讲。还有一个高频问题cuda多版本安装。很多人一台机器上要跑不同的项目有的要 CUDA 10.2有的要 11.8有的要 12.1。这时候你不需要反复卸载重装CUDA 是支持多版本共存的通过软链接和环境变量切换就行。具体操作后面有详细步骤。最后说一个反直觉的点不是所有 N 卡都能装 CUDA。你的显卡算力Compute Capability必须 3.0太老的卡比如 GT 610 这种就别折腾了。另外AMD 显卡原生跑 CUDA 这件事网上确实有一些方案但本质上都是通过转译层或者重编译实现的性能和兼容性都有折损不是“完美运行”这个后面会专门聊。2. 驱动安装从 nvidia-smi 到黑屏救援的完整流程2.1 驱动安装前的系统检查装驱动之前先确认三件事显卡型号、操作系统版本、是否已有旧驱动残留。显卡型号用lspci | grep -i nvidiaLinux或者设备管理器Windows看。操作系统版本决定了你下载哪个驱动包Ubuntu 22.04 和 20.04 的驱动包是不一样的Windows 10 和 11 也不一样。旧驱动残留是重装系统后最容易忽略的问题。如果你之前装过驱动哪怕是在“程序和功能”里卸载了注册表里可能还有残留。Linux 下更麻烦/etc/modprobe.d/下面可能有blacklist-nouveau.conf之类的文件/usr/lib/modules/下面可能有旧的内核模块。这些不清干净新驱动装上去大概率会冲突表现就是开机黑屏或者循环登录。提示Linux 下装驱动前先禁用 nouveau 开源驱动。这个驱动是内核自带的会和 N 卡官方驱动抢设备。禁用方法是在/etc/modprobe.d/blacklist-nouveau.conf里写入blacklist nouveau和options nouveau modeset0然后sudo update-initramfs -u更新内核镜像重启生效。2.2 Windows 下的驱动安装Windows 下装驱动相对简单但有几个细节要注意。首先下载驱动的时候选“Game Ready Driver”还是“Studio Driver”如果你主要是跑深度学习选 Studio Driver它的稳定性更好更新频率低不会在你跑实验的时候突然弹窗让你重启。Game Ready Driver 更新太频繁有时候新驱动反而会引入兼容性问题。安装的时候选“自定义安装”然后勾选“执行清洁安装”。这个选项会把旧驱动彻底清掉再装新的虽然会多花几分钟但能避免很多玄学问题。如果你不勾有时候旧驱动的残留文件会导致nvidia-smi能跑但 CUDA 程序报错。安装完成后打开命令行输入nvidia-smi如果能看到显卡信息表格说明驱动装好了。表格里重点看三个东西Driver Version、CUDA Version、GPU 名称。Driver Version 是你装的驱动版本CUDA Version 是这个驱动支持的最高 CUDA 版本GPU 名称确认是你那张卡。2.3 Linux 下的驱动安装Linux 下装驱动有三种方式apt 包管理器、runfile 安装包、conda 安装。我推荐 apt 方式最省心。以 Ubuntu 为例先ubuntu-drivers devices看看系统推荐哪个版本然后sudo apt install nvidia-driver-550这样装。装完重启nvidia-smi能出来就行。runfile 方式适合需要精细控制的情况比如你要装特定版本的驱动或者 apt 源里的版本太老。下载.run文件后先sudo telinit 3进入文本模式然后sudo sh NVIDIA-Linux-x86_64-xxx.run运行安装程序。安装过程中会问你要不要装 32 位兼容库、要不要更新 X 配置一般选是就行。装完sudo telinit 5回到图形界面。conda 方式最不推荐虽然conda install cudatoolkit看起来很方便但它装的是 CUDA 运行时库不是完整驱动。而且 conda 的 CUDA 版本往往滞后和系统驱动的兼容性也不好。除非你是在没有 root 权限的集群上跑否则别用 conda 装 CUDA。注意Linux 下装完驱动如果黑屏别慌。按 CtrlAltF2 进文本终端登录后sudo apt remove --purge nvidia-*把驱动卸干净然后重新装。如果连文本终端都进不去就用 Live USB 启动chroot 进去修。这个坑我踩过三次每次都是因为没禁用 nouveau。2.4 WSL2 下的驱动逻辑WSL2 的驱动安装和纯 Linux 完全不同。你不需要在 WSL2 里装任何显卡驱动只需要在 Windows 宿主机上装好驱动就行。WSL2 通过一个叫/usr/lib/wsl/lib的目录直接访问宿主机的驱动库。你可以在 WSL2 里ls /usr/lib/wsl/lib看看里面应该有libcuda.so之类的文件。如果你在 WSL2 里跑nvidia-smi报错说找不到命令先检查/usr/lib/wsl/lib在不在 PATH 里。通常这个目录会被自动加到 PATH但有时候需要手动加。在~/.bashrc里加一行export PATH$PATH:/usr/lib/wsl/lib就行。WSL2 里装 CUDA Toolkit 的步骤和纯 Linux 一样但注意不要装驱动。NVIDIA 官方文档里 WSL2 的 CUDA 安装指南明确说了只装 Toolkit不装 Driver。如果你手贱装了 Linux 驱动WSL2 会直接挂掉只能重装 WSL2。3. CUDA Toolkit 安装nvcc 不是内部命令的根治方案3.1 Toolkit 和 Driver 的区别这是最容易被混淆的一对概念。Driver 是让操作系统认识显卡的Toolkit 是让你写 CUDA 程序的。nvidia-smi来自 Drivernvcc来自 Toolkit。你可以只装 Driver 不装 Toolkit这时候nvidia-smi能用但nvcc不能用。你也可以装多个版本的 Toolkit但 Driver 只有一个。nvcc 不是内部或外部命令,也不是可运行的程序,或批处理文件这个报错99% 的原因是 Toolkit 没装或者装了但没加到 PATH。Windows 下 Toolkit 默认装在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin你需要把这个路径加到系统环境变量 PATH 里。Linux 下默认装在/usr/local/cuda-12.4/bin同样要加到 PATH。验证 Toolkit 装好没有用nvcc -V。注意是大写 V小写 v 在某些版本上不显示版本号。输出应该类似nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2024 NVIDIA Corporation Built on Wed_Apr_17_19:19:55_PDT_2024 Cuda compilation tools, release 12.4, V12.4.131 Build cuda_12.4.r12.4/compiler.34097967_0如果nvcc -V报错但nvidia-smi正常那就是 Toolkit 的问题不是驱动的问题。别去折腾驱动浪费时间。3.2 Linux 下安装 CUDA Toolkit以 Ubuntu 22.04 装 CUDA 12.4 为例。先去 NVIDIA 官网下载页选 Linux → x86_64 → Ubuntu → 22.04 → runfile (local)。下载下来是个.run文件大概 4 个 G。安装命令sudo sh cuda_12.4.1_550.54.15_linux.run运行后会先解压然后弹出安装界面。这里有个关键操作在安装选项里把 Driver 那一项的勾去掉。因为你已经装过驱动了Toolkit 自带的驱动版本可能和你系统里的不一样勾上会覆盖掉现有驱动容易出问题。只勾 CUDA Toolkit 和 Samples 就行。安装完成后配置环境变量。在~/.bashrc里加export PATH/usr/local/cuda-12.4/bin:$PATH export LD_LIBRARY_PATH/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH然后source ~/.bashrc生效。nvcc -V能出版本号就说明装好了。3.3 Windows 下安装 CUDA ToolkitWindows 下更简单下载.exe安装包双击运行。安装类型选“自定义”然后取消勾选 Driver 组件只留 CUDA 和 Development 相关的。安装路径默认就行别改改了容易出玄学问题。装完后检查环境变量。系统属性 → 高级 → 环境变量看看Path里有没有C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin。如果没有手动加。加完重启命令行nvcc -V验证。提示Windows 下如果nvcc -V报错先检查是不是装了多个版本的 Toolkit。多个版本共存时PATH 里哪个版本在前面nvcc就用哪个。你可以通过调整 PATH 顺序来切换版本。3.4 WSL2 下安装 CUDA ToolkitWSL2 下装 Toolkit 和纯 Linux 几乎一样但有几个坑。首先WSL2 的 Ubuntu 版本要和 Toolkit 支持的版本匹配。Ubuntu 22.04 装 CUDA 12.x 没问题但装 CUDA 11.x 可能会缺依赖。其次WSL2 下不要装 Driver前面说过了。安装命令和 Linux 一样但注意 WSL2 的/usr/local/cuda软链接可能不会自动创建。装完后手动建一下sudo ln -s /usr/local/cuda-12.4 /usr/local/cuda然后配 PATH 和 LD_LIBRARY_PATH。WSL2 下nvidia-smi是从/usr/lib/wsl/lib调用的和 Toolkit 的路径不冲突两个都能用就对了。3.5 多版本共存与切换一台机器上装多个 CUDA 版本是常态。Linux 下所有版本都装在/usr/local/下面比如cuda-11.8、cuda-12.1、cuda-12.4。/usr/local/cuda是个软链接指向当前默认版本。切换版本就是改软链接sudo rm /usr/local/cuda sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda然后重新source ~/.bashrc。但注意如果你的 PATH 里写的是/usr/local/cuda-12.4/bin这种硬编码路径改软链接没用得改 PATH。所以我建议 PATH 里写/usr/local/cuda/bin这样切换软链接就能全局生效。Windows 下多版本切换靠改 PATH 顺序。把想用的版本的bin目录挪到最前面就行。但 Windows 下有个坑nvcc和cudart的版本可能不一致因为 PATH 里可能混了不同版本的库。建议用where nvcc确认一下实际调用的是哪个。4. cuDNN 安装版本匹配比安装本身更重要4.1 cuDNN 是什么为什么需要它cuDNN 是 NVIDIA 的深度神经网络加速库全称 CUDA Deep Neural Network library。PyTorch、TensorFlow 这些框架底层调用的卷积、池化、归一化等操作都是通过 cuDNN 实现的。没有 cuDNN框架也能跑但速度会慢很多因为会退回到 CUDA 的通用实现。cuDNN 的版本必须和 CUDA 版本严格对应。CUDA 11.8 对应 cuDNN 8.9.xCUDA 12.1 对应 cuDNN 8.9.x 或 9.xCUDA 12.4 对应 cuDNN 9.x。装错了版本框架启动时就会报Could not locate cudnn_ops_infer64_8.dll或者libcudnn.so.8: cannot open shared object file这种错。4.2 Linux 下安装 cuDNNcuDNN 的下载需要注册 NVIDIA 开发者账号这个没办法跳过。下载下来是个 tar 包解压后里面有三个目录include、lib64、lib。把它们拷贝到 CUDA 的对应目录tar -xzvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 sudo chmod ar /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*验证安装cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2能看到版本号就对了。注意如果你装了多个 CUDA 版本cuDNN 要分别拷贝到每个版本的目录下。比如 CUDA 11.8 和 12.4 都装了那 cuDNN 8.9 要拷到 11.8 的目录cuDNN 9.x 要拷到 12.4 的目录。别偷懒只拷一个。4.3 Windows 下安装 cuDNNWindows 下下载的是 zip 包解压后同样有bin、include、lib三个目录。把bin里的 dll 拷到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bininclude里的 h 文件拷到...\includelib里的 lib 文件拷到...\lib\x64。验证方法写个简单的 CUDA 程序调用 cuDNN或者直接跑框架看能不能启动。更简单的方法是看C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin下面有没有cudnn64_8.dll或者cudnn64_9.dll。4.4 版本匹配速查表CUDA 版本推荐 cuDNN 版本适用框架11.28.1.0TensorFlow 2.5.011.88.9.7PyTorch 2.0-2.212.18.9.7 / 9.1PyTorch 2.112.49.1PyTorch 2.3这张表建议存下来装之前对一遍。我见过太多人拿着 TensorFlow 2.5.0 去配 CUDA 12.4然后问为什么跑不起来。不是 CUDA 的问题是 TensorFlow 2.5.0 编译的时候就是针对 CUDA 11.2 的你给它 12.4 它不认识。5. 验证与排查从 nvidia-smi 到实际跑通框架5.1 三层验证法装完驱动、Toolkit、cuDNN 之后别急着跑框架先做三层验证。第一层nvidia-smi。能出表格Driver Version 和 CUDA Version 都正常说明驱动没问题。第二层nvcc -V。能出编译器版本说明 Toolkit 没问题。第三层编译并运行一个 CUDA Samples。CUDA Toolkit 自带 Samples在/usr/local/cuda/samples或者C:\ProgramData\NVIDIA Corporation\CUDA Samples。进去make一下然后跑./bin/x86_64/linux/release/deviceQuery。如果输出Result PASS说明 CUDA 运行时没问题。如果cuda samples找不到可能是安装的时候没勾 Samples 组件。Linux 下可以单独下载 Samples 包Windows 下重新运行安装程序勾上就行。5.2 框架层面的验证PyTorch 验证import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.cuda.get_device_name(0)) print(torch.version.cuda)如果torch.cuda.is_available()返回 False先检查 PyTorch 版本和 CUDA 版本是否匹配。PyTorch 官网有版本对照表装之前对一下。如果返回 True 但get_device_name报错可能是驱动和 Toolkit 版本不匹配。TensorFlow 验证import tensorflow as tf print(tf.__version__) print(tf.config.list_physical_devices(GPU))TensorFlow 2.5.0 比较特殊它要求 CUDA 11.2 cuDNN 8.1而且对驱动版本也有要求。如果你驱动太新反而可能不兼容。这时候要么降驱动要么升 TensorFlow。5.3 常见报错速查报错信息原因解决方案nvcc: command not foundToolkit 没装或 PATH 没配装 Toolkit配 PATHCUDA error: no kernel image显卡算力和编译目标不匹配重新编译指定正确的算力libcudnn.so.8: cannot opencuDNN 没装或版本不对装对应版本的 cuDNNCUDA out of memory显存不够减小 batch size或换卡driver version insufficient驱动版本太低升级驱动nvidia-smi: command not found驱动没装或 PATH 没配装驱动配 PATH5.4 独家避坑技巧第一个坑别用最新版。CUDA 和驱动都不是越新越好。新版本刚出来的时候框架往往还没适配你装了就是当小白鼠。等三个月等框架官方说支持了再升。第二个坑WSL2 里别装驱动。这个前面说过了但真的有人不信邪装了之后 WSL2 直接挂掉数据全丢。WSL2 的驱动是宿主机管的你只需要装 Toolkit。第三个坑多版本共存时LD_LIBRARY_PATH 的顺序很重要。如果你 PATH 里是 CUDA 12.4但 LD_LIBRARY_PATH 里是 CUDA 11.8 的 lib64那运行时链接的库是 11.8 的会出各种诡异错误。确保 PATH 和 LD_LIBRARY_PATH 指向同一个版本。第四个坑cuDNN 的版本号要看仔细。cuDNN 8.9.7 和 8.9.6 可能只差一个小版本但对应的 CUDA 版本可能不一样。下载的时候看清楚文件名里的cuda11还是cuda12。第五个坑4060 Ti 的算力是 8.9编译的时候要指定-archsm_89。如果你用默认的编译选项可能编译出来的二进制在 4060 Ti 上跑不了。PyTorch 官方编译的版本一般会包含多个算力但你自己编译的时候要注意。6. 进阶话题多版本管理、迁移与特殊场景6.1 CUDA 迁移到新机器换机器的时候CUDA 环境不能直接拷贝因为驱动和硬件绑定。正确的迁移步骤是新机器先装驱动然后装相同版本的 Toolkit 和 cuDNN最后把项目代码拷过去。如果项目里有编译好的二进制可能需要重新编译因为算力目标可能不一样。如果新机器显卡算力比旧机器低编译的时候要降低算力目标。比如旧机器是 A100算力 8.0新机器是 4060 Ti算力 8.9那编译的时候要包含 sm_89。反过来如果旧机器算力低新机器算力高一般没问题因为 CUDA 是向前兼容的。6.2 llama.cpp 与 CUDA 的兼容性cuda llama.cpp non compatible这个问题最近问的人很多。llama.cpp 的 CUDA 后端对版本比较敏感它要求 CUDA 11.7 以上但太新的版本也可能有问题。如果你用 CUDA 12.4 编译 llama.cpp 报错试试降到 12.1 或者 11.8。编译 llama.cpp 的 CUDA 后端make LLAMA_CUDA1如果报错说找不到 CUDA检查CUDA_PATH环境变量。llama.cpp 的 Makefile 会读这个变量没设的话它找不到 Toolkit。6.3 AMD 显卡跑 CUDA 的真相网上有些文章说 AMD 显卡可以“完美运行 CUDA”这个说法不太准确。AMD 显卡跑 CUDA 本质上是通过 HIPHeterogeneous-compute Interface for Portability或者 ZLUDA 这样的转译层实现的。HIP 是 AMD 自己的 CUDA 替代方案代码需要移植ZLUDA 是直接转译 CUDA 二进制但兼容性和性能都有折损。如果你手头只有 AMD 卡想跑 CUDA 程序最靠谱的方案是用 ROCm。ROCm 是 AMD 的官方计算平台PyTorch 和 TensorFlow 都有 ROCm 版本。但 ROCm 支持的显卡型号有限不是所有 AMD 卡都能用。而且 ROCm 的安装比 CUDA 还麻烦坑更多。6.4 SLI 与 CUDA 的关系sli cuda这个组合现在很少见了。SLI 是 NVIDIA 的多卡渲染技术主要用于游戏和 CUDA 计算没关系。CUDA 的多卡计算靠的是 NCCL 和 CUDA 的 P2P 访问不需要 SLI 桥接器。如果你有多张卡想跑 CUDA直接装驱动和 Toolkit 就行不用管 SLI。多卡 CUDA 编程的时候用cudaSetDevice切换设备用cudaMemcpyPeer做卡间通信。NCCL 库封装了这些操作做分布式训练的时候直接用 NCCL 就行。6.5 驱动更新与 CUDA 版本升级驱动更新和 CUDA 升级是两件事。驱动更新不影响已安装的 Toolkit但可能影响 Toolkit 的可用性。比如你驱动从 550 升到 555CUDA 12.4 的 Toolkit 可能还能用但 CUDA 11.8 的可能就出问题了因为新驱动可能不再支持老版本的运行时。升级 CUDA 版本的时候建议先备份当前环境然后装新版本测试没问题再删旧版本。别直接覆盖安装出了问题回滚很麻烦。7. 实操心得与常见问题实录7.1 我的标准安装流程以 Ubuntu 22.04 RTX 4060 Ti CUDA 12.4 为例我的标准流程是这样的装系统更新到最新sudo apt update sudo apt upgrade禁用 nouveau写 blacklist 文件update-initramfs -u重启装驱动sudo apt install nvidia-driver-550重启验证驱动nvidia-smi能出表格装 Toolkit下载 runfile安装时取消 Driver 勾选配环境变量PATH 和 LD_LIBRARY_PATH验证 Toolkitnvcc -V能出版本装 cuDNN下载对应版本拷贝文件验证 cuDNN检查头文件和库文件装 PyTorchpip install torch --index-url https://download.pytorch.org/whl/cu124验证 PyTorchtorch.cuda.is_available()返回 True这套流程我跑了不下二十遍每次都能一次成功。关键点就三个禁用 nouveau、取消 Driver 勾选、版本严格匹配。7.2 那些年我踩过的坑第一个坑在 WSL2 里装了 Linux 驱动WSL2 直接挂掉重装花了两个小时。教训是 WSL2 只装 Toolkit不装 Driver。第二个坑cuDNN 版本装错TensorFlow 启动时报Could not create cudnn handle。查了半天才发现是 cuDNN 8.9 配了 CUDA 11.2应该用 cuDNN 8.1。教训是版本对照表要常备。第三个坑多版本共存时 PATH 和 LD_LIBRARY_PATH 不一致PyTorch 能 import 但跑不了卷积。查了一天才发现是运行时链接了错误版本的库。教训是 PATH 和 LD_LIBRARY_PATH 必须指向同一个版本。第四个坑驱动自动更新后 CUDA 挂了。Ubuntu 的 unattended-upgrades 会自动更新驱动有时候新驱动和老 Toolkit 不兼容。教训是关掉自动更新或者锁定驱动版本。7.3 常见问题速查表问题排查思路解决方案nvidia-smi 报错驱动没装或版本不对重装驱动nvcc 找不到Toolkit 没装或 PATH 没配装 Toolkit配 PATH框架检测不到 GPU驱动/Toolkit/cuDNN 版本不匹配对照版本表逐一检查编译 CUDA 程序报错算力目标不对指定正确的 -arch显存不足batch size 太大减小 batch size多卡训练报错NCCL 没装或版本不对装 NCCL配环境变量WSL2 里 nvidia-smi 找不到PATH 没包含 /usr/lib/wsl/lib加 PATH驱动更新后 CUDA 挂了新驱动不兼容老 Toolkit降驱动或升 Toolkit7.4 最后的经验分享装 CUDA 这件事说难不难说简单也不简单。核心就一句话版本匹配。驱动、Toolkit、cuDNN、框架四个东西的版本必须对得上。对上了一路畅通对不上各种报错。我的建议是装之前先花十分钟查版本对照表别急着下载。下载的时候看清楚文件名别下错平台。安装的时候仔细看选项别一路 Next。装完做三层验证别急着跑框架。还有别追新。CUDA 12.4 刚出的时候我就装了结果 PyTorch 还不支持又降回 12.1。等三个月等生态跟上了再升省心省力。最后备份环境。装之前把当前环境用conda env export或者pip freeze导出来出了问题能快速回滚。这个习惯我养了五年救了我无数次。
阅读完成 · 觉得有帮助?