创建新的虚拟环境
conda create –n 虚拟环境名字 python=版本
conda activate 虚拟环境名字
找到cuda版本,cmd用管理员方式运行
nvidia-smi
进入官网
找到对应的命令语句输入则可安装pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
检查是否安装成功
pip list
或者conda list
import torch
print(torch.cuda.is_available())