PyTorch

一、查看PyTorch版本及其对应CUDA版本

PyTorch官网：https://pytorch.org/#community-module

二、环境安装

1. 安装对应版本的cuDNN与CUDA
2. Python安装：Miniconda安装方法

3. 安装好Miniconda后使用命令创建Python环境

4. 创建名为tf_env的Python3.9环境
   conda create -n my_env python=3.9

5. 进入Python环境
   conda activate my_env

6. .pip安装PyTorch
   在Python环境中输入该命令
   安装最新版
   pip3 install torch torchvision torchaudio

7. 指定CUDA版本
   pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
8. 检测释放安装成功,返回为true则安装成功
   python -c "import torch;torch.cuda.is_available();"
   ps：如下载资源过慢可调整pip使用清华源

设置清华源，加速安装
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
安装最新版
pip3 install torch torchvision torchaudio
指定CUDA版本安装
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://d