Linux安装pytorch如何查看对应的cuda本版(两种cuda版本应该如何选择?)
于 2023-10-06 13:20:38 首次发布
博客主要围绕Linux安装pytorch时如何查看对应的cuda版本及两种cuda版本的选择展开。介绍了查看当前实际安装cuda版本的命令,MMCV编译依据此版本;还提到查看NVIDIA驱动版本及匹配Cuda版本的命令。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.6
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理


7836





