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

nr7wwzry  于 2022-11-09  发布在  其他
关注(0)|答案(2)|浏览(220)

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

以上是nvidia --versionnvidia-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
lsmd5eda

lsmd5eda1#

我想你也需要cudatoolkit
conda安装是最简单的:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

或使用cuda 10.1

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
ovfsdjhp

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,就这样。

相关问题