安装
-
创建目录
C:\CUDA\v11.8
, 安装CUDA 11.8 到该目录下: https://developer.nvidia.com/cuda-11-8-0-download-archive -
安装pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
正文完