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

自动安装当前最新版本。

 

 

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值