0
点赞
收藏
分享

微信扫一扫

下好torch和cuda包之后python中gpu和cuda还是显示Flase的问题

灵魂跑者 2022-04-27 阅读 56

首先要确定你torch包是能用的,不报错

第二,

去conda清华源找到对应的pytorch gpu版本(cuda版本)

打开 Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

比如你下的是python3.8,cuda11.1,你就去找那个包

直接ctrl+f 搜文件名即可

例如 pytorch-1.11.0-py3.7_cuda11.1_cudnn8_0.tar.bz2

之后的话找到你的虚拟环境,进入之后直接找到下载文件根目录

输入 

conda install --offline  pytorch-1.11.0-py3.7_cuda11.1_cudnn8_0.tar.bz2

之后pycharm检查会发现已成功,这个问题是因为你之前下载文件的时候他会自动给你推荐cpu的torch包,gpu根本没装上

举报

相关推荐

0 条评论