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