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/

    分割线:

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

  • 相关阅读:
    HDU 4868 Information Extraction(2014 多校联合第一场 H)
    Transformations 方块转换
    catalan 数——卡特兰数(转)
    算法分析与设计——矩阵连乘问题
    算法设计与分析——多边形游戏(DP)
    蓝桥杯算法训练 最大最小公倍数
    codeforces 518B. Tanya and Postcard
    并查集
    高精度的进制转换
    线段树(转)
  • 原文地址:https://www.cnblogs.com/jisongxie/p/10055523.html
Copyright © 2011-2022 走看看