别用conda命令安装,就用pip命令安装,前提是安装了对应的CUDA和CUDNN。
比如现在(2024.12)安装CUDA11.8对应的pytorch官网给出的命令是这个。你的电脑的CUDA可能不是11.8,安装的时候就把这行命令的最后的cu118换成你的CUDA版本。也不用考虑向下兼容不兼容,直接安装符合自己CUDA的版本就ookk。
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
更正:cuda是向下兼容,pytorch按照cuda版本强适配。
所以如果你的cuda版本不是pytorch安装命令中的cu***强匹配的,可以降低后面的cu***来安装,比如你的电脑上cuda版本是12.0,pytorch安装命令向下最近只支持cu11.8,那就向下兼容为11.8。