conda create -n pytorch python=3.10
conda activate pytorch
conda install -c conda-forge ipykernel
python -m ipykernel install --user --name pytorch
查看cuda版本
nvidia-smi
上官网找对应的安装代码
//这是我的安装代码
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia