pytorch torch.cuda.is_available()在Ubuntu 20.04上安装适当版本的CUDA和CuDNN后返回FALSE

nr7wwzry 于 2022-11-09 发布在其他

关注(0)|答案(2)|浏览(213)

我应该说在Z490主板上安装Ubuntu 20.04是相当棘手的。

以上是nvidia --version和nvidia-smi的屏幕截图
我能够完成所有CUDA和CUDNN安装。
我用了所有这些。但GPU是没有检测到与pytorch。

pip3 install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

pip install torch torchvision

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

非常感谢。谁来帮我把这个弄出来。

whereis cuda
cuda: /usr/lib/cuda /usr/include/cuda.h

whereis nvcc
nvcc: /usr/bin/nvcc /etc/nvcc.profile /usr/share/man/man1/nvcc.1.gz

但是在一些论坛中我看到路径是/usr/local/cuda
除此之外，我的~/.bashrc还包含以下内容：

export LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH

pytorch

来源：https://stackoverflow.com/questions/63797147/torch-cuda-is-available-retuns-false-after-installing-appropriate-versions-of

2条答案

按热度按时间

lsmd5eda1#

我想你也需要cudatoolkit。
从conda安装是最简单的：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

或使用cuda 10.1

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

赞(0）回复(0）举报 2022-11-09

ovfsdjhp2#

此问题已解决。

$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C03sv00001043sd000085ABbc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106 [GeForce GTX 1060 6GB]
driver   : nvidia-driver-390 - distro non-free
driver   : nvidia-driver-435 - distro non-free
driver   : nvidia-driver-440 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

最初，在安装nvidia驱动程序时，nvidia-driver-450不可用。但CUDA 11工具包是基于nvidia-driver-450构建的。为此，我们需要使用PPA存储库安装Nvidia Beta驱动程序。
使用，

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt install nvidia-driver-450

剩下的过程是一样的。安装CUDA和CUDA工具包，CuDNN和bammmm，就这样。

赞(0）回复(0）举报 2022-11-09

我来回答

pytorch torch.cuda.is_available()在Ubuntu 20.04上安装适当版本的CUDA和CuDNN后返回FALSE

2条答案

相关问题

热门标签

最新问答