cuda/cudnn版本查看并安装合适版本pytorch

1.CUDA版本查看：

cat /usr/local/cuda/version.txt

我的版本:10.0.138
2.cudnn版本查看：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

根据以上返回，可以看出cudnn版本为7.6.5

进入pytorch官网
选择合适自己合适版本conda下载即可（网速问题emmmm）