1
0
Files
AI-learning/README.md
2025-01-04 00:28:28 +08:00

705 B

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

tensorflow

python

# win and linux
pip install tensorflow
#mac apple silicon
pip install tensorflow-macos
pip install tensorflow-metal

c++