小编Kai*_*ing的帖子

当我尝试使用 OpenCV 函数 cv2.cvtColor() 转换 RGB 图像时，我得到了一个类似绿色的图像。

我已经将 OpenCV 读取的原始图像转换为 RGB 格式，再次使用 cv2.cvtColor() 将其转换为灰度，并尝试使用 pyplot.imshow() 函数显示它。

image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
plt.imshow(image)

我在 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。怎么了？提前致谢！