之前装了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/
分割线:
------------------------------------------------------------------------------------------------------------------------