我在 PyTorch 上运行 CNN。 torch.cuda.is_available() 函数返回 false 并且未检测到 GPU。不过,我可以使用 GPU 运行 Keras 模型。这是我的系统信息:
- 操作系统:Ubuntu 18.04.3
- Python 3.7.3(康达)
- 显卡:GTX1080Ti
- Nvidia 驱动程序:430.50
当我检查nvidia-smi时,输出显示CUDA版本是10.1。但是,nvcc -V 命令告诉我它是 CUDA 9.1。
我从官方网站下载了NVIDIA-Linux-x86_64-430.50.run并使用命令行安装。我使用官方网站推荐的以下命令行安装了 CUDA 10.1:
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run
我通过 pip install 安装了 PyTorch。怎么了?提前致谢!
默认的 Pytorch 1.2 包依赖于 CUDA 10.0,但您有 CUDA 9.1。的输出nvidia-smi
只是告诉您 GPU 支持的最大 CUDA 版本,nvcc
给出系统上安装的 CUDA。看来您的 CUDA 10.1 安装不成功。
除了 CUDA 10.0 之外,Pytorch 还支持 CUDA 9.2,并且我发现为 CUDA 10.0 编译的 Pytorch 包也适用于 CUDA 10.1。因此,您可以将 CUDA 安装升级到 9.2 并使用以下命令安装 Pytorch CUDA 9.2 软件包
pip3 install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
或者获取 CUDA 10.1 的有效安装。有详细的Linux说明here. https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html(请注意,在安装新的 CUDA 之前,您可能必须删除以前安装的 CUDA。)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)