python - 即使安装了 CUDA 和带有 cuda 的 pytorch，Pytorch cuda 也不可用。怎么修？-6ren

python - 即使安装了 CUDA 和带有 cuda 的 pytorch，Pytorch cuda 也不可用。怎么修？

转载作者：行者123 更新时间：2023-12-05 05:58:25

我正在尝试在使用 WSL2 的 Windows 机器上将 pytorch 与我的 GPU (RTX 3070) 结合使用，但即使我遵循了 Nvidia 指南 (https://docs.nvidia.com/cuda/wsl-user-guide/index.html#abstract)，我也无法让它工作。

nvidia-smi.exe 输出:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 471.41       Driver Version: 471.41       CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name            TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ... WDDM  | 00000000:0A:00.0  On |                  N/A |
|  0%   40C    P5    12W / 220W |   1815MiB /  8192MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1384    C+G   Insufficient Permissions        N/A      |
|    0   N/A  N/A      1628    C+G   ...dows\System32\WWAHost.exe    N/A      |
|    0   N/A  N/A      3172    C+G   ...y\ShellExperienceHost.exe    N/A      |
|    0   N/A  N/A      5940    C+G   ...lPanel\SystemSettings.exe    N/A      |
|    0   N/A  N/A      6360    C+G   C:\Windows\explorer.exe         N/A      |
|    0   N/A  N/A      7280    C+G   ...artMenuExperienceHost.exe    N/A      |
|    0   N/A  N/A      7732    C+G   ...5n1h2txyewy\SearchApp.exe    N/A      |
|    0   N/A  N/A      8256    C+G   ...ekyb3d8bbwe\YourPhone.exe    N/A      |
|    0   N/A  N/A      8780    C+G   ...nputApp\TextInputHost.exe    N/A      |
|    0   N/A  N/A     10032    C+G   ...perience\NVIDIA Share.exe    N/A      |
|    0   N/A  N/A     10732    C+G   ...hyper\app-3.0.2\Hyper.exe    N/A      |
|    0   N/A  N/A     10804    C+G   ...kyb3d8bbwe\Calculator.exe    N/A      |
|    0   N/A  N/A     10852    C+G   ...in7x64\steamwebhelper.exe    N/A      |
|    0   N/A  N/A     12180    C+G   ...ge\Application\msedge.exe    N/A      |
|    0   N/A  N/A     13880    C+G   ...icrosoft VS Code\Code.exe    N/A      |
|    0   N/A  N/A     14724    C+G   ...b3d8bbwe\WinStore.App.exe    N/A      |
+-----------------------------------------------------------------------------+

nvcc -v 输出:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Wed_Jun__2_19:15:15_PDT_2021
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0

dpkg -l | grep nvidia 输出:

ii  libnvidia-cfg1-470:amd64             470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA binary OpenGL/GLX configuration library
ii  libnvidia-common-470                 470.57.02-0ubuntu0.20.04.1        all          Shared files used by the NVIDIA libraries  
rc  libnvidia-compute-450:amd64          450.51.05-0ubuntu1                amd64        NVIDIA libcompute package
rc  libnvidia-compute-465:amd64          465.27-0ubuntu0.20.04.2           amd64        NVIDIA libcompute package
ii  libnvidia-compute-470:amd64          470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA libcompute package
ii  libnvidia-decode-470:amd64           470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA Video Decoding runtime libraries    
ii  libnvidia-encode-470:amd64           470.57.02-0ubuntu0.20.04.1        amd64        NVENC Video Encoding runtime library       
ii  libnvidia-extra-470:amd64            470.57.02-0ubuntu0.20.04.1        amd64        Extra libraries for the NVIDIA driver      
ii  libnvidia-fbc1-470:amd64             470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA OpenGL-based Framebuffer Capture runtime library
ii  libnvidia-gl-470:amd64               470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA OpenGL/GLX/EGL/GLES GLVND libraries 
and Vulkan ICD
ii  libnvidia-ifr1-470:amd64             470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA OpenGL-based Inband Frame Readback runtime library
ii  libnvidia-ml-dev                     10.1.243-3                        amd64        NVIDIA Management Library (NVML) development files
rc  nvidia-compute-utils-450             450.51.05-0ubuntu1                amd64        NVIDIA compute utilities
ii  nvidia-compute-utils-470             470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA compute utilities
ii  nvidia-cuda-dev                      10.1.243-3                        amd64        NVIDIA CUDA development files
ii  nvidia-cuda-doc                      10.1.243-3                        all          NVIDIA CUDA and OpenCL documentation       
ii  nvidia-cuda-gdb                      10.1.243-3                        amd64        NVIDIA CUDA Debugger (GDB)
ii  nvidia-cuda-toolkit                  10.1.243-3                        amd64        NVIDIA CUDA development toolkit
rc  nvidia-dkms-450                      450.51.05-0ubuntu1                amd64        NVIDIA DKMS package
ii  nvidia-dkms-470                      470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA DKMS package
ii  nvidia-driver-470                    470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA driver metapackage
rc  nvidia-kernel-common-450             450.51.05-0ubuntu1                amd64        Shared files used with the kernel module   
ii  nvidia-kernel-common-470             470.57.02-0ubuntu0.20.04.1        amd64        Shared files used with the kernel module   
ii  nvidia-kernel-source-470             470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA kernel source package
ii  nvidia-opencl-dev:amd64              10.1.243-3                        amd64        NVIDIA OpenCL development files
ii  nvidia-prime                         0.8.16~0.20.04.1                  all          Tools to enable NVIDIA's Prime
ii  nvidia-profiler                      10.1.243-3                        amd64        NVIDIA Profiler for CUDA and OpenCL        
ii  nvidia-settings                      450.51.05-0ubuntu1                amd64        Tool for configuring the NVIDIA graphics driver
ii  nvidia-utils-470                     470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA driver support binaries
ii  nvidia-visual-profiler               10.1.243-3                        amd64        NVIDIA Visual Profiler for CUDA and OpenCL 
ii  screen-resolution-extra              0.18build1                        all          Extension for the nvidia-settings control panel
ii  xserver-xorg-video-nvidia-470        470.57.02-0ubuntu0.20.04.1        amd64        NVIDIA binary Xorg driver

dpkg -l | grep cuda 输出:

ii  cuda                                 11.4.0-1                          amd64        CUDA meta-package
ii  cuda-11-4                            11.4.0-1                          amd64        CUDA 11.4 meta-package
ii  cuda-cccl-11-4                       11.4.43-1                         amd64        CUDA CCCL
ii  cuda-command-line-tools-11-4         11.4.0-1                          amd64        CUDA command-line tools
ii  cuda-compiler-11-4                   11.4.0-1                          amd64        CUDA compiler
rc  cuda-cudart-11-0                     11.0.194-1                        amd64        CUDA Runtime native Libraries
ii  cuda-cudart-11-4                     11.4.43-1                         amd64        CUDA Runtime native Libraries
ii  cuda-cudart-dev-11-4                 11.4.43-1                         amd64        CUDA Runtime native dev links, headers     
ii  cuda-cuobjdump-11-4                  11.4.43-1                         amd64        CUDA cuobjdump
ii  cuda-cupti-11-4                      11.4.65-1                         amd64        CUDA profiling tools runtime libs.
ii  cuda-cupti-dev-11-4                  11.4.65-1                         amd64        CUDA profiling tools interface.
ii  cuda-cuxxfilt-11-4                   11.4.43-1                         amd64        CUDA cuxxfilt
ii  cuda-demo-suite-11-4                 11.4.43-1                         amd64        Demo suite for CUDA
ii  cuda-documentation-11-4              11.4.43-1                         amd64        CUDA documentation
ii  cuda-driver-dev-11-4                 11.4.43-1                         amd64        CUDA Driver native dev stub library        
ii  cuda-gdb-11-4                        11.4.55-1                         amd64        CUDA-GDB
ii  cuda-libraries-11-4                  11.4.0-1                          amd64        CUDA Libraries 11.4 meta-package
ii  cuda-libraries-dev-11-4              11.4.0-1                          amd64        CUDA Libraries 11.4 development meta-package
ii  cuda-memcheck-11-4                   11.4.43-1                         amd64        CUDA-MEMCHECK
ii  cuda-nsight-11-4                     11.4.43-1                         amd64        CUDA nsight
ii  cuda-nsight-compute-11-4             11.4.0-1                          amd64        NVIDIA Nsight Compute
ii  cuda-nsight-systems-11-4             11.4.0-1                          amd64        NVIDIA Nsight Systems
ii  cuda-nvcc-11-4                       11.4.48-1                         amd64        CUDA nvcc
ii  cuda-nvdisasm-11-4                   11.4.43-1                         amd64        CUDA disassembler
ii  cuda-nvml-dev-11-4                   11.4.43-1                         amd64        NVML native dev links, headers
ii  cuda-nvprof-11-4                     11.4.43-1                         amd64        CUDA Profiler tools
ii  cuda-nvprune-11-4                    11.4.43-1                         amd64        CUDA nvprune
ii  cuda-nvrtc-11-4                      11.4.50-1                         amd64        NVRTC native runtime libraries
ii  cuda-nvrtc-dev-11-4                  11.4.50-1                         amd64        NVRTC native dev links, headers
ii  cuda-nvtx-11-4                       11.4.43-1                         amd64        NVIDIA Tools Extension
ii  cuda-nvvp-11-4                       11.4.43-1                         amd64        CUDA Profiler tools
ii  cuda-repo-ubuntu2004-11-0-local      11.0.2-450.51.05-1                amd64        cuda repository configuration files        
ii  cuda-repo-wsl-ubuntu-11-4-local      11.4.0-1                          amd64        cuda repository configuration files        
ii  cuda-runtime-11-4                    11.4.0-1                          amd64        CUDA Runtime 11.4 meta-package
ii  cuda-samples-11-4                    11.4.43-1                         amd64        CUDA example applications
ii  cuda-sanitizer-11-4                  11.4.54-1                         amd64        CUDA Sanitizer
rc  cuda-toolkit-11-0                    11.0.2-1                          amd64        CUDA Toolkit 11.0 meta-package
ii  cuda-toolkit-11-4                    11.4.0-1                          amd64        CUDA Toolkit 11.4 meta-package
ii  cuda-toolkit-11-4-config-common      11.4.43-1                         all          Common config package for CUDA Toolkit 11.4.
ii  cuda-toolkit-11-config-common        11.4.43-1                         all          Common config package for CUDA Toolkit 11.
ii  cuda-toolkit-config-common           11.4.43-1                         all          Common config package for CUDA Toolkit.    
ii  cuda-tools-11-4                      11.4.0-1                          amd64        CUDA Tools meta-package
rc  cuda-visual-tools-11-0               11.0.2-1                          amd64        CUDA visual tools
ii  cuda-visual-tools-11-4               11.4.0-1                          amd64        CUDA visual tools
ii  libcudart10.1:amd64                  10.1.243-3                        amd64        NVIDIA CUDA Runtime Library
ii  nvidia-cuda-dev                      10.1.243-3                        amd64        NVIDIA CUDA development files
ii  nvidia-cuda-doc                      10.1.243-3                        all          NVIDIA CUDA and OpenCL documentation       
ii  nvidia-cuda-gdb                      10.1.243-3                        amd64        NVIDIA CUDA Debugger (GDB)
ii  nvidia-cuda-toolkit                  10.1.243-3                        amd64        NVIDIA CUDA development toolkit

我安装的pytorch版本:

torch             1.9.0+cu111
torchaudio        0.9.0
torchvision       0.10.0+cu111

但是pytorch告诉我cuda不可用:

Python 3.8.10 (default, Jun  2 2021, 10:49:15) 
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False

有谁知道我是否安装了我需要的所有软件包？我尝试了互联网上发布的许多解决方案，但仍然无效。

最佳答案

我的环境是(带有 NVIDIA GTX 1080Ti 的 Ubuntu 20.04):

$ nvidia-smi | grep CUDA
| NVIDIA-SMI 470.74       Driver Version: 470.74       CUDA Version: 11.4     |

$ nvcc -V               
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Aug_15_21:14:11_PDT_2021
Cuda compilation tools, release 11.4, V11.4.120
Build cuda_11.4.r11.4/compiler.30300941_0

安装 CUDA 和 cuDNN 后，我运行了从 https://pytorch.org/get-started/locally/ 获取的 followwing 命令带有选项(PyTorch Stable 1.9.1、Linux、Pip、Python、CUDA 11.1)。

pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

查找包:

(test-py38) ➜  ~ conda list | grep torch
_pytorch_select           0.1                       cpu_0    defaults
pytorch                   1.8.1           cpu_py38h60491be_0    defaults
torch                     1.9.1+cu111               <pip>
torchaudio                0.9.1                     <pip>
torchvision               0.10.1+cu111              <pip>

然后:

$ python           
Python 3.8.11 (default, Aug  3 2021, 15:09:35) 
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

关于python - 即使安装了 CUDA 和带有 cuda 的 pytorch，Pytorch cuda 也不可用。怎么修？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/68493965/

文章推荐： javascript - 将 div 翻译成像里程表一样的行为

文章推荐： wordpress - 如何在 docker 上创建多个 wordpress 站点

javascript - 带有 li & ul 的多级下拉列表，带有 PHP 数组循环，带有 CSS 和 jQuery
我想使用 li 和 ul 制作一个多级下拉列表，以便显示我博客中按年和月排序的所有文章。我希望我的下拉菜单看起来像 Google Blogspot 下拉菜单: 这是我的 CSS 和 HTML 代码 u
c++ - 带有 gmp 的代码块，带有 << 运算符和 mp*_class 的段错误
我在 Win 7 64 机器上将 CodeBlocks 与 gcc 4.7.2 和 gmp 5.0.5 结合使用。开始使用 gmpxx 后，我看到一个奇怪的段错误，它不会出现在 +、- 等运算符中，但
javascript - 带有 Tern 的 CodeMirror - 带有 Javascript Intellisense 的自定义 "Types"
我正在使用 tern 为使用 CodeMirror 运行的窗口提供一些增强的智能感知，它工作正常，但我遇到了一个问题，我想添加一些自定义“types”，可以这么说，这样下拉列表中它们旁边就有图标了。我
带有 PC 作为 USB 主机的 Android ADK，带有 libusb，批量传输错误
我正在尝试让我的 PC 成为 Android 2.3.4 设备的 USB 主机，以便能够在不需要实际“附件”的情况下开发 API。为此，我需要将 PC 设置为 USB 主机和“设备”(在我的例子中是运
php - 带有 IIS 的 ASP.NET VS 带有 Apache 的 PHP
很难说出这里要问什么。这个问题模棱两可、含糊不清、不完整、过于宽泛或夸夸其谈，无法以目前的形式得到合理的回答。如需帮助澄清此问题以便重新打开，visit the help center . 关闭 9
apache - 带有 Apache/Tomcat 多个 DNS 的反向代理，带有 SSL 的单个 Web 应用程序
我在设置服务器方面几乎是个新手，但遇到了一个问题。我有一个 Ubuntu 16.04 VPS 并安装了 Apache2 和 Tomcat7。我正在为 SSL 使用 LetsEncrypt 和 Cert
google-maps - 如何有效地将 Google Maps API V2(带有 MapFragment、Google Map、Marker 等)用于使用 API V1(带有 MapView、GeoPoint 等)的旧应用程序，
我在一个基于谷歌地图的项目上工作了超过 6 个月。我使用的是 Google Maps API V1 及其开发人员 API key 。当我尝试发布应用程序时，我了解到 Google API V1 已被弃
带有@property的Python对象来听写
我是 Python 的新手，所以如果我对一些简单的事情感到困惑，请原谅。我有一个这样的对象: class myObject(object): def __init__(self):
带有#号的javascript对象变量
这个问题已经有答案了: How can I access object properties containing special characters? (2 个回答) 已关闭 9 年前。我正在尝
带有@media的纵向模式和陆地模式的CSS
我有下面的 CSS。我想要的是一种流体/液体(因为缺乏正确的术语)css。我正在为移动设备开发，当我改变模式时从纵向 View 到陆地 View ，我希望它流畅。现在的图像在陆地 View 中效
带有@语法的python装饰器参数
我正在尝试使用可以接受参数的缓存属性装饰器。我查看了这个实现:http://www.daniweb.com/software-development/python/code/217241/a-cac
带有-1作为第三个参数的python双冒号
这个问题在这里已经有了答案: Understanding slicing (36 个答案) 关闭 6 年前。以a = [1,2,3,4,5]为例。根据我的直觉，我认为 a[::-1] 与 a[0:
带有--where子句的mysqldump不起作用
mysqldump -t -u root -p mytestdb mytable --where=datetime LIKE '2014-09%' 这就是我正在做的事情，它会返回: mysqldum
python - 带有 while 循环的销售税计算器
我正在制作销售税计算器，除了总支付金额部分外，其他一切都正常。在我的程序中，我希望能够输入一个数字并获得该项目的税额我还希望能够获得支付的总金额，包括交易中的税金。到目前为止，我编写的代码完成了所有这
带有-hwaccel_output_format的FFMPEG Hwaccel错误
关闭。这个问题不符合Stack Overflow guidelines .它目前不接受答案。我们不允许在 Stack Overflow 上提出有关通用计算硬件和软件的问题。您可以编辑问题，使其成为
带有 Airflow 的电子邮件
我是否必须进行任何额外的设置才能让 apache-airflow 在任务失败时向我发送电子邮件。我的配置文件中有以下内容(与默认值保持不变): [email] email_backend = airf
excel - 带有 $ 的内置字符串函数
这个问题在这里已经有了答案: What does the $ symbol do in VBA? (5 个回答) 3年前关闭。使用返回字符串(如 Left)的内置函数有什么区别吗？或使用与 $ 相同
.net - 带有.NET库的VB6
我有一个用VB6编写的应用程序，我需要使用一个用.NET编写的库。有什么方法可以在我的应用程序上使用该库吗？谢谢最佳答案这取决于。您可以控制.NET库吗？如果是这样，则可以修改您的库，以便可以
raku - 带有 ^ 的类方法名称没有被正确调用
当我创建一个以 ^ 开头的类方法时，我尝试调用它，它给了我一个错误。 class C { method ^test () { "Hi" } } dd C.new.test; Too m
带有 Material 设计的Angularjs无法实例化模块ngMaterial
我已经使用 bower 安装了 angularjs 和 materialjs。凉亭安装 Angular Material 并将“ngMaterial”注入(inject)我的应用程序，但出现此错误。

行者123

个人简介

我是一名优秀的程序员,十分优秀！

作者热门文章

滴滴打车优惠券免费领取

全站热门文章

首页

博学

6Ren·AI

商城

python - 即使安装了 CUDA 和带有 cuda 的 pytorch，Pytorch cuda 也不可用。怎么修？