CUDA相关问题

之前装了GPU驱动后，再装了CUDA 9.0，再装了cuDNN，并且对样例mnistCUDNN进行执行，显示“Test passed!"通过。但是倒忘了有没有测试CUDA是否安装成功。驱动也有测试，使用nvidia-smi，可打印GPU信息。今天下午，遇到一个问题后，重装了驱动，想着用不用重装CUDA这些。测试了一下CUDA，发现好像没装成功？？

此时电脑上已经装了pytorch，于是想看看变量能否被传输到显卡上。

1、程序运行前，显卡占用为198MiB。

2、执行如下语句后：

import torch

dev = torch.device('cuda')

a = torch.randn(200, 200, 200).to(dev)

显卡占用变为472MiB。后面还试了更大的张量，结果也显示占用了更多的显存。因此说明张量有被传输到显卡上。

但是不明白为什么nvcc -V不存在。于是我按照提示，执行sudo apt install nvidia-cuda-toolkit，又babala安装了一些东西，安装完成后再次输入nvcc -V，便有输出了。

但此时，却发现nvidia-smi失败了。

 上网查了一下，好像重启就能解决，因为刚安装了一些东西。于是就重启了，果然奏效。

所以，CUDA应该是有被正确安装上了的。

 

cuda的官方安装指导：https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/

分割线：

 ------------------------------------------------------------------------------------------------------------------------