Ubuntu 20.04 CUDA for torch无法通过pytorch使用

m1m5dgzv 于 2023-06-23 发布在其他

关注(0)|答案(1)|浏览(161)

问题：

1.安装了nvida驱动程序后是否重新安装pytorch

pytorch是否有CUDA版本和无CUDA版本？

描述：

我从ubuntu 20.4安装了N-videa和cUDA。但仍然无法从pytorch获取cUDA，请参阅python3脚本

(torch) anaconda $ python3 -c  "import torch; print(f'CUDA:{torch.cuda.is_available()} CUDAs:{torch.cuda.device_count()}')"
CUDA:False CUDAs:0

检查n-vidia驱动和CUDA

（torch）anaconda $ nvidia-smi
2019年10月15日星期四20：18：57 2023
+----------------------------------------------------------------------------------+
| NVIDIA-SMI 530.30.02驱动程序版本：530.30.02 CUDA版本：12.1|
|- ———————————————————————————————————————-+-————————————————————-+-————————————————————-+

(torch) anaconda $  /usr/local/cuda-12/bin/nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Mon_Apr__3_17:16:06_PDT_2023
Cuda compilation tools, release 12.1, V12.1.105
Build cuda_12.1.r12.1/compiler.32688072_0

(torch) anaconda $ lspci -k|grep VGA -A2
01:00.0 VGA compatible controller: NVIDIA Corporation Device 2484 (rev a1)
        DeviceName: VGA
        Subsystem: ASUSTeK Computer Inc. Device 87bc
        Kernel driver in use: nvidia

pytorch

来源：https://stackoverflow.com/questions/76482495/ubuntu-20-04-cuda-for-torch-unavailable-by-pytorch

1条答案

按热度按时间

ttp71kqs1#

经检查，我的pytorch只有CPU版本。所以我创建了新的conda env并再次安装了pytorch。
$ conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
然后再检查一遍，它现在正在工作！

$ python3 -c  "import torch; print(f'CUDA: {torch.zeros(1).cuda()} ; TorchVer: {torch.__version__}')"
CUDA: tensor([0.], device='cuda:0') ; TorchVer: 2.0.1

$ python3 -c  "import torch; print(f'CUDA: {torch.cuda.is_available()} CUDAs:{torch.cuda.device_count()}')"
CUDA:True CUDAs:1

赞(0）回复(0）举报 2023-06-23

我来回答

Ubuntu 20.04 CUDA for torch无法通过pytorch使用

1条答案

相关问题

热门标签

最新问答