一、安装PyTorch(torch)
torch相当于是PyTorch,torch利用python操作cuda,cuda利用驱动操作gpu。
根据CUDA版本安装pytorch。
cmd查看cuda版本:
nvidia-smi
去pytorch官网查看对应版本依赖,我cuda是12.1所以使用下面代码安装
# CUDA 12.1
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=12.1 -c pytorch -c nvidia
二、下载pyskl代码
https://github.com/kennymckormick/pyskl
三、环境安装
conda create -n pyskl python=3.8
conda activate pyskl