1.查看pytorch版本
终端输入
import torch
print(torch.__version__)
2.查看cuda版本
方法1:
cat /usr/local/cuda/version.txt
方法2:
nvcc -V
3.点击这里选择合适的libtorch版本
请注意,在Linux上提供了两种类型的libtorch二进制文件:一种是用GCC pre-cx11 ABI编译的,另一种是用GCC-cx11 ABI编译的,应该根据系统使用的GCC ABI进行选择。
参考文档
https://pytorch.org/docs/master/cpp_index.html#packaging-for-c