赞
踩
用Anaconda安装 PyTorch 2.0.1,GPU版,CUDA版本为11.7:
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
但是torch可用,cuda不可用:
import torch
#torch是否可用;
print(torch.__version__) #2.0.1
#cuda是否可用;
print(torch.cuda.is_available()) #False
看网上的解决方案,大多需要确定是否安装为GPU版本,以及CUDA版本是否一致(见下一节),但我这都没问题。
打开命令提示符(Win+R -> cmd)输入:
nvcc -V
出来的最后两行就是版本号
网上还有其他方法查看CUDA,但是看的不是一个东西(详见另一篇博客),以这个为准就完事了。
安装更低版本PyTorch即可(2.0.0及以下),原因没整明白,离谱。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。