检查当前环境下的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 中
- 找cudnn_version.h
- 或找 cudnn.h
- 使用命令 cat {.h路径} | grep CUDNN_MAJOR 查看版本(一般是版本数字)
- 查看当前cuda toolkit版本: nvcc -V
- 查看当前 cuda driver 版本: nvidia-smi
对照版本之间是否支持