仅作为记录，大佬请跳过。

查看linux的所有cuda版本

用

cd /usr/local
ls

查看所有的cuda版本

当前运行的cuda版本

在/usr/local路径下，用

stat cuda

展示

参考

传送门

两种cuda

nvidia-smi：无用cuda

nvcc -V：有用cuda

传送门

测试cuda是否能使用

import torch
torch.cuda.is_available()
# 返回True 接着用下列代码进一步测试
torch.zeros(1).cuda()

参考：传送门

关于cuda的知识

查看gpu的算力（即nvidia的算力）

nvidia-smi
nvidia-smi -a

nvidia官网的算力查询

查看pytorch支持的cuda算力

import torch
torch.cuda.get_arch_list()

Out[2]: [‘sm_37’, ‘sm_50’, ‘sm_60’, ‘sm_70’]