如何查看Docker中cuda的版本?

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

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

conda activate xxx

2. 进入python环境

python

3. 查看cuda版本

import torch
print(torch.version.cuda)

在这里插入图片描述

你可能感兴趣的:(Ubuntu系统,docker,python,linux)