首页自学conda环境下print(torch.cuda.is_available()) False

conda环境下print(torch.cuda.is_available()) False

adminadmin时间2024-06-28 19:24:24分类自学浏览47

情景是:明明pip install torch>=1.10 安装成功了,但conda环境下运行


print(torch.cuda.is_available())


却得到


False


后来通过


print(torch.__version__)
确实是安装提CPU版本而非GPU版本



找到以前另一个安装笔记使用以下安装解决


conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia


但正确的处理方法应该是Pip手工安装 ,指定清楚,比如以下demo

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2
速度慢还可以用迅雷下载实际文件后 直接安装。


参考自:https://blog.csdn.net/qq_46126258/article/details/112708781


本文《conda环境下print(torch.cuda.is_available()) False》jingyan2/57.html

与《conda环境下print(torch.cuda.is_available()) False》

0
0
收藏0
德展健康:控股股东提议2亿元至3亿元回购股份

    未登录用户 回复需填写必要信息
    请先 登录 再评论,若不是会员请先 注册