我使用anaconda进行安装pytorch时候,由于我使用的命令是
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
就报错了
我刚开始试着把pip3变成pip,还是不行,然后翻看这篇文章,
后来想着是不是不能用pip,要用conda。
然后去官网查看了以下conda命令
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
输入这行就可以安装了