【一】Pytorch安装

一、Pytorch安装

conda 安装

只用conda环境的建议安装mini conda即可，轻量级；

anaconda：https://www.anaconda.com/download/success

miniconda：

win:
  curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Windows-x86_64.exe -o miniconda.exe
  start /wait "" miniconda.exe /S
  del miniconda.exe
  
linux:
  mkdir -p ~/miniconda3
  wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
  bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
  rm -rf ~/miniconda3/miniconda.sh
  
mac:
  mkdir -p ~/miniconda3
  curl https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh -o ~/miniconda3/miniconda.sh
  bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
  rm -rf ~/miniconda3/miniconda.sh

创建conda环境

#conda create –n 虚拟环境名字 python=版本
conda create -n torch1.13_py38 python=3.8
#conda create –n 虚拟环境名字 python=版本 –c 镜像地址

#镜像地址：
清华镜像:         https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
北京外国语大学镜像  https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
阿里巴巴镜像:      http://mirrors.aliyun.com/anaconda/pkgs/main
#激活环境：
conda activate torch1.13_py38

安装包

1. conda安装方式

   #conda install xxx
   conda install opencv_python
   #指定版本
   conda install opencv_python==4.6.0
   #下载慢，从镜像网站安装
   conda install opencv_python -c 镜像地址
   #镜像地址：
   镜像名      |镜像地址 – pytorch, torchvision, torchaudio
   清华镜像    |https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
   北外镜像    |https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/
   阿里镜像    |http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
   南大镜像    |https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/
   
   #配置永久的镜像地址：
   conda config --show
   conda config --add channels 镜像地址
   conda config --remove channels 镜像地址
   

2. pip安装方式（推荐）

   #pip install xxx
   #pip3 install xxx
   pip3 install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
   #配置pip永久镜像地址：
   pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
   
   

CUDA 安装

cuda版本概念：

linux安装CUDA Toolkit：

官网：https://developer.nvidia.com/cuda-toolkit-archive

1. 选择对应的版本（不高于nvidia-smi的cuda版本即可）；
2. 选择 win 或 linux（x86_64）；
3. 选择ubuntu/wsl-ubuntu（wsl子系统）；
4. 选择ubuntu版本20.04/22.04；
5. 选择安装类型runfile（local），下载安装；

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
sudo sh cuda_12.2.0_535.54.03_linux.run

6. accept cuda 安装界面，回车消除其他安装选项（仅保留toolkit）；
8. 
   ![外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传](https://img-home.csdnimg.cn/images/20230724024159.png?origin_url=%2FUsers%2Fyangchangjie%2FDocuments%2Fmd%2Fcudatoolkit.png&pos_id=img-dJQWISbj-1719931889464

配置环境变量：

# 配置用户环境变量
vim ~/.bashrc

# 在该文件的最末尾添加两行
export PATH=/path/to/cuda11.7/bin:$PATH
export LD_LIBRARY_PATH=/path/to/cuda11.7/lib64:$LD_LIBRARY_PATH

# :wq保存文件，source命令更新用户环境
source ~/.bashrc

验证nvcc -V

win安装：

PyTorch安装

官网：https://pytorch.org/get-started/previous-versions/

可以在线安装，也可以下载下安装

如，cuda11.7 安装torch1.13

pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117