如何查看Docker中cuda的版本？

最近想在服务器上的一个docker上添加新的环境，但是忘了是什么版本的cuda了，记录下小坑。

1. 激活一个docker已经存在的环境

conda activate xxx
1

2. 进入python环境

python
1

3. 查看cuda版本

import torch
print(torch.version.cuda)
1
2