python+cuda+torch安装

编辑于2023.11.23

1.python安装

• python版本：3.10

（1）python官网：https://www.python.org/

（2）python安装选项：
- Customize Installation：自定义安装
- Add python.exe to PATH：勾选

2.Anaconda安装

2.1 Anaconda官网：https://www.anaconda.com/download

2.2 pip配置
（1）pip配置国内源-清华源

• /用户/[user] -> 新建文件夹pip -> 新建文件pip.ini -> 输入：

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

[install]
use-mirrors =true
mirrors =https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host =pypi.tuna.tsinghua.edu.cn

（2）pip临时更换源
pip install [name] -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

（3）查看pip源
pip config list

2.3 conda配置
（1）conda配置国内源-清华源

• 进入终端cmd：conda config --set show_channel_urls yes -> 生成.condarc文件
• 打开：/用户/[user]/.condarc，输入文字：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
show_channel_urls: true
auto_activate_base: false

（2）查看conda源：conda config --get

3.PyCharm安装

（1）PyCharm官网：https://www.jetbrains.com/pycharm/

4.cuda安装

• cuda版本：11.8

（1）显卡驱动下载
https://www.nvidia.cn/Download/index.aspx?lang=cn

（2）查看CUDA所需的最低版本显卡驱动
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

（3）CUDA下载官方网址：版本11.8
https://developer.nvidia.com/cuda-toolkit-archive

（4）查看CUDA安装成功：nvcc -V

（5）查看显卡信息：nvidia-smi

5.cudnn安装

（1）cudnn官网：https://developer.nvidia.com/rdp/cudnn-archive

（2）选择对应版本下载：CUDA v11.x - cuDNN v8.9.4

（3）cudnn安装：解压[下载文件]，文件全部拷贝至CUDA文件夹下
- CUDA默认安装文件夹：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

（4）查看cudnn安装成功
- 进入/extras/demo_suite/…：shfit+右键-powershell
- ./deviceQuery.exe
- ./bandwidthTest.exe
- ‘Result=PASS’：cudnn安装成功

6.torch安装

6.1 torch所需文件：
（1）torch：核心功能
（2）torchvision：图像处理包
（3）torchaudio：语言处理包

6.2 torch安装
（1）torch官网
- 官网：https://pytorch.org/
- 下滑-选择：[Stable2.1.1]、[windows]、[python]、[CUDA 11.8]
- 命令：pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
- 若网速好，选择以上命令行；若网速不好，选择以下步骤

（2）下载网址：https://download.pytorch.org/whl/cu118
- torch -> 搜索’cu118’
- 选择：torch-2.1.0；cu118；cp310；win
- 使用[迅雷下载]速度快

（3）安装：进入powershell，
- pip install [torch-2.1.x.whl] torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

6.3 查看pytorch安装成功
cmd：

• python
• import torch
• torch.__version__
• torch.cuda.is_available()
• 若True，则安装成功

6.4 torch卸载
- pip uninstall torch torchvision torchaudio

PS：SD-xformers：目前最高支持torch版本-2.1.0