查看虚拟环境cuda和cudnn版本命令

在使用Python进行深度学习开发，并利用CUDA和cuDNN进行加速计算时，通常会在安装好PyTorch或TensorFlow这样的库之后，在虚拟环境中检查CUDA和cuDNN的版本。以下是几个常用的命令：

1. **查看CUDA版本**:
- 对于Linux系统（如Ubuntu）：

     CUDA_HOME=$(dirname $(which nvidia-smi)) # 获取nvidia-smi的位置并取上一级目录
     . $CUDA_HOME/bin/activate # 激活cuda环境变量
     nvcc --version # 显示CUDA版本信息

- 对于Windows用户：
打开命令提示符，输入 `nvcc --version`。

2. **查看cuDNN版本**:
- Linux:

     conda list | grep cudatoolkit # 如果使用Conda管理环境，可以搜索cuDNN
     python -c "import torch; print(torch.backends.cudnn.version())" # 或者直接导入torch库查看

- Windows:

     pip show cupy-cuda100 # 如果使用cupy库，pip show torch-cu100 或 torch-nightly-cu100

记得先激活相关的虚拟环境，因为不同的环境可能安装了不同的CUDA和cuDNN版本。