1
0
This commit is contained in:
2025-01-11 13:56:41 +08:00
parent c0d101ca1b
commit 81367980c1
13 changed files with 610 additions and 40 deletions

View File

@@ -1,47 +1,30 @@
# pytorch study
# AI-Learning
## BASE ENV
```shell
conda create -n pt python=3.10 -y
conda activate pt
```
## MAC
```shell
conda create -n ail python=3.10 -y
conda activate ail
# 安装 pytorch v1.12版本已经正式支持了用于mac m1芯片gpu加速的mps后端
conda install pytorch::pytorch torchvision torchaudio -c pytorch -y
# tensorflow mac apple silicon
pip install tensorflow-macos
pip install tensorflow-metal
pip install -r requirements.txt
```
## linux
```shell
conda create -n ail-tf python=3.9 -y
conda create -n ail-pt python=3.10 -y
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
# tensorflow 需要cudnn支持
pip install tensorflow
pip install -r requirements.txt
```
## Jupyter
```shell
# env
pip install jupyter
```
```shell
jupyter notebook
```
## tensorflow
### python
```shell
# win and linux
pip install tensorflow
```
```shell
#mac apple silicon
pip install tensorflow-macos
pip install tensorflow-metal
```
### c++