torch获取不到cuda可能原因

检查当前环境下的pytorch是不是cuda版本(还是仅cpu)？

• conda list
• pip list
  有关conda和pip的关系，可以看这篇：Conda与Pip的融合
  总结就是：
  • conda是包管理器、环境管理器；pip是python的包安装工具
  • conda下可以下载pip，并管理pip。conda环境下使用pip下载的包在…/anaconda3/envs/name/lib/pythonXX/site-packages/下
  • 不在conda下的pip，通过which pip可以找到路径
    另外：conda创建环境时，如果指定python=3.x, 就会在该anaconda/env中安装python，否则会使用环境变量找到的第一个python

检查cudnn, cuda toolkit(即使用的cuda编译器)， cuda driver版本是否兼容

• 直接搜索 cudnn cuda toolkit support matrix找到官网页面
• 查看当前cudnn版本：
  • /usr/include 或 /usr/local/cuda(自己的cuda路径)/include 中
  1. 找cudnn_version.h
  2. 或找 cudnn.h
  • 使用命令 cat {.h路径} | grep CUDNN_MAJOR 查看版本（一般是版本数字）
• 查看当前cuda toolkit版本： nvcc -V
• 查看当前 cuda driver 版本： nvidia-smi
  对照版本之间是否支持