pytorch安装巨坑–清华镜像
使用GPU处理数据会报错cuda与pytorch不匹配。
因为清华镜像上下载的包不是GPU版,是CPU,对你没看错,是CPU版的
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
解决方法
- 卸载pytorch
conda uninstall pytorch
- 使用非镜像源的安装。这里我的pip没有改变镜像源,所以使用pip指令下载
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
注:这里不需要翻墙也可以下载
可以使用以下代码测试
python
import torch
print(torch.cuda.is_available())