zoukankan      html  css  js  c++  java
  • 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/

    分割线:

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

  • 相关阅读:
    宏任务、微任务
    类和模块
    每日日报
    每日日报
    每日日报
    每日日报
    每日日报
    每日日报
    每日日报
    每日日报
  • 原文地址:https://www.cnblogs.com/jisongxie/p/10055523.html
Copyright © 2011-2022 走看看