libtorch支持的CuDA版本
时间: 2025-01-09 17:36:34 浏览: 110
### LibTorch 支持的 CUDA 版本
对于不同版本的 PyTorch 和对应的 LibTorch,所支持的 CUDA 版本有所不同。具体来说:
- **PyTorch 1.8.x** 及其相应的 LibTorch 支持 CUDA 10.2 和 CUDA 11.1[^1]。
- **PyTorch 1.9.x** 及其相应的 LibTorch 支持 CUDA 10.2 和 CUDA 11.3。
- **PyTorch 1.10.x** 及其相应的 LibTorch 支持 CUDA 10.2 和 CUDA 11.3。
由于提到的是 `torch1.10.1` 是 CPU 版本,因此如果需要使用带有 CUDA 加速的功能,则应安装对应版本的 GPU 版本 PyTorch 或者 LibTorch,并确保该版本与本地已有的 CUDA 工具包相兼容。
为了确认特定版本的 LibTorch 所能适配的具体 CUDA 版本号,在下载页面通常会有详细的说明文档提供给开发者作为参考依据。另外也可以通过官方 GitHub 发布的日志以及发行说明来获取更详尽的信息。
```cpp
// 示例:验证当前环境中使用的CUDA版本(C++)
#include <iostream>
#include <cuda.h>
int main() {
int version;
cudaError_t error = cudaDriverGetVersion(&version);
if (error != cudaSuccess) {
std::cout << "Failed to get CUDA driver version." << std::endl;
return -1;
}
std::cout << "Current CUDA Driver Version is: " << version / 1000 << "." << (version % 1000)/10 << std::endl;
}
```
阅读全文
相关推荐


















