torch.cuda.is_available()返回false的解决办法
作者:XD / 发表: 2020年4月27日 01:51 / 更新: 2020年5月22日 02:14 / 编程笔记 / 阅读量:9520
显卡是没问题的,主要还是cudatoolkit版本的问题。
nvidia-smi 查看cuda版本
然后复制这一行就搞定了
conda install pytorch torchvision cudatoolkit=xxx(选择对应的版本) -c pytorch
相关标签
显卡是没问题的,主要还是cudatoolkit版本的问题。
nvidia-smi 查看cuda版本
然后复制这一行就搞定了
conda install pytorch torchvision cudatoolkit=xxx(选择对应的版本) -c pytorch
本站现有博文258篇,共被浏览413141次
本站已经建立1978天!