Pytorch 与 Tensorflow安装问题

Pytoroch 出现  no CUDA-capable device is detected : 

1. RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:7

一般是由运行的脚本中设置了CUDA_VISIBLE_DEVICES="3"。可以通过该命令可以看到需要使用编号为3的显卡,关于显卡编号一般是从0开始编号的,第一块编号为0,第二块编号为1。虽然指定使用第二块显卡,不过我的计算机上面只有一块显卡,所以这样会提示上面的报错信息。将显卡指定行改为如下CUDA_VISIBLE_DEVICES="0"改完可以正常运行了

2. 关于tensorflow-GPU版本的安装:

使用conda安装:

conda install tensorflow-gpu

自动安装当前最新版本。

 

 

 

你可能感兴趣的:(机器学习)