之前装了GPU驱动后,再装了CUDA 9.0,再装了cuDNN,并且对样例mnistCUDNN进行执行,显示“Test passed!"通过.但是倒忘了有没有测试CUDA是否安装成功.驱动也有测试,使用nvidia-smi,可打印GPU信息.今天下午,遇到一个问题后,重装了驱动,想着用不用重装CUDA这些.测试了一下CUDA,发现好像没装成功?? 此时电脑上已经装了pytorch,于是想看看变量能否被传输到显卡上. 1.程序运行前,显卡占用为198MiB. 2.执行如下语句后: import t…