添加环境变量
- 如果 CUDA 已安装,但 nvcc 无法识别,你需要将 CUDA 的 bin 目录添加到你的 PATH 环境变量中。假设 CUDA 安装在 /usr/local/cuda,你可以使用以下命令更新你的 ~/.bashrc 文件:
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
2.然后,重新加载 .bashrc 文件:
source ~/.bashrc
3.检测mmcv的版本
python -c "import mmcv; print(mmcv.__version__)"