pytorch gpu安装 cuda.is_available=False
·
显卡驱动对应cuda是12.4，但是这个没有对应的pytorch
安装cuda11.3，与pytorch 版本不兼容
cuda 建议安装11.8稳定版本
然后直接用官方命令暗转pytorch即可
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
python=3.8
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
3
0- 0
已为社区贡献1条内容
所有评论(0)