docker 查看pytorch 和cuda版本

在Docker环境中查看PyTorch和CUDA版本通常需要先启动容器，并在容器内部执行命令。以下是步骤：

1. **确保你已经有一个包含PyTorch和CUDA的Docker镜像**：这通常是通过运行一个特定的Dockerfile构建的，比如`nvidia/pytorch` 或者 `nvcr.io/nvidia/pytorch:latest`。

2. **打开终端并进入你的Docker容器**：如果你已经通过命令行运行了容器，你可以使用`docker exec -it <container_name> /bin/bash` 进入容器的shell环境，其中 `<container_name>` 是你的容器名。

3. **检查CUDA版本**：在bash shell中输入：

   nvidia-smi

如果CUDA安装成功，这个命令会显示GPU的信息，包括CUDA的版本。

4. **检查PyTorch版本**：在shell中输入：

   python -c "import torch; print(torch.__version__)"

这将打印出你当前使用的PyTorch库的版本。

5. **关闭终端，退出容器**：当你完成查看后，可以使用 `exit` 退出shell。