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

首先要确定你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根本没装上

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值