conda list
查看pytorch版本,一定要安装到对应的路径下,别一直在base下安装!
pytorch版本太低,不支持3090的算力,需要重新安装pytorch
官网:https://pytorch.org/get-started/locally/
CUDA version查询:nvidia-smi
把命令复制到终端下载
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
cuda安装成功print(torch.cuda.is_available())却输出为false的原因有二:
1.版本号不匹配
2.pytorch不小心安装为cpu版,与cuda适配的pytorch应该为gpu版本不知道什么原因,用cuda的命令安装,老是会安成CPU版本的,卸载CPU版本的pytorch,此时换pip命令试一下
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113
错误CPU版本的pytorch
安装过程会显示: