问题:
只有pytorch-cpu,无法使用如下命令安装pytorch-gpu
conda install pytorch-cuda=11.6
解决:
查看cuda版本,比如11.6
nvidia-smi
根据cuda版本使用pip3安装torch
pip3 install torch --index-url https://download.pytorch.org/whl/cu116
问题:
只有pytorch-cpu,无法使用如下命令安装pytorch-gpu
conda install pytorch-cuda=11.6
解决:
查看cuda版本,比如11.6
nvidia-smi
根据cuda版本使用pip3安装torch
pip3 install torch --index-url https://download.pytorch.org/whl/cu116