命令行输入nvidia-smi,查看驱动信息
nvidia-smi
看右上角cuda版本
coda 11.6的安装版本
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
命令行输入nvidia-smi,查看驱动信息
nvidia-smi
看右上角cuda版本
coda 11.6的安装版本
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge