由于在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了

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐