出现这个问题,不仅是 `nvcc -V`会显示command not found,`nvidia-smi`同样也会显示
解决方法如下:
1)这里首先转换到CUDA所在位置,一般是在这个位置
cd /usr/local
2)打开、编辑环境变量的配置文件
vim ~/.bashrc
在末尾加上以下内容,最后":wq"退出保存即可
# cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
3)刷新文件,使其生效
source ~/.bashrc
4)测试一下
nvcc -V
接下来是一些常见查看版本的命令:
nvidia-smi #查看显卡使用情况
nvidia-smi -L #查看显卡型号
使用python文件查看常见命令
import torch #查看torch版本
print(torch.__version__) #注意是双下划线import torch
print(torch.cuda.is_available())#如果cuda可用,则返回Truetorch.cuda.device_count() # 查看GPU数量
torch.cuda.is_available() # 查看是否有可用GPU
torch.cuda.current_device() #查看当前使用的显卡,返回的是显卡的序号