pytorch study
BASE ENV
conda create -n pt python=3.10 -y
conda activate pt
MAC
# 安装 pytorch v1.12版本已经正式支持了用于mac m1芯片gpu加速的mps后端
conda install pytorch::pytorch torchvision torchaudio -c pytorch -y
pip install -r requirements.txt
linux
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
pip install -r requirements.txt
Jupyter
# env
pip install jupyter
jupyter notebook