【无标题】解决pycharm中torch.cuda.is_available()为false的问题
·
由于在pycharm中import torch之后,发现torch.cuda.is_available()一直为false,就去网上搜了很多教程,最终成功解决
首先配置anaconda环境,在pycharm中新创建一个conda解释器

我们这里创建的环境名字叫3,然后进入anaconda prompt中,输入
conda env list

发现了我们创建的环境3,然后
conda activate 3
进入我们的环境,首先确保cuda是正确安装的,我们安装的版本为10.2

然后直接输入
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
就开始自动下载和cuda对应的torch和torchvision库了,下载完之后在pycharm中就可以发现
torch.cuda.is_available()变为true了
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)