Update pytorch.md

Author: ZhikangNiu

src/ai/pytorch.md

@@ -4,40 +4,40 @@
 https://pytorch.org/get-started/previous-versions/
 ### pip 
 #### 安装最新版
-```
+```shell
 pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
 ```
 #### 指定版本(preffered)
 创建名为pytorch310的虚拟环境
-```
+```shell
 conda create -n pytorch310 python==3.10
 ```
 激活环境
-```
+```shell
 conda activate pytorch310
 ```
 设置清华源，加速安装
-```
+```shell
 pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
 ```
 指定版本安装
-```
+```shell
 pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 ```
 ### conda
 添加清华镜像源
-```
+```shell
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
 conda config --set show_channel_urls yes
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
 ```
 #### 安装最新版
-```
+```shell
 conda install pytorch torchvision torchaudio cudatoolkit=10.2
 ```
 #### 指定版本
-```
+```shell
 conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4 cudatoolkit=10.2
 ```