PyTorch 环境搭建¶

更新日期：20220817

1. 安装和配置 Python¶

1.1. 安装 Python¶

下载页面下载 3.9 的版本。默认下载为 x86，请手动选择 x64 版本

安装时：

勾选 Add Python 3.x to PATH 。
并选择自定义安装位置。
勾选 Install for all users 。
选择需要的安装位置。
点击安装。

1.2. 设置镜像¶

设置镜像【有些时候镜像会有问题，则需要清除对应文件。】：

sudo vi ~/.pip/pip.conf 中添加或修改，Windows 中为用户目录中创建 pip/pip.ini 文件。

[global]
index-url = https://mirrors.aliyun.com/pypi/simple/

[install]
trusted-host=mirrors.aliyun.com

1.3. 更新 pip 并安装 virtualenv¶

更新 pip 执行： C:\Tools\Python39\python.exe -m pip install --upgrade pip

pip install virtualenv
virtualenv --version

1.4. virtualenv 常用命令¶

这里我们的虚拟环境全部存放到 C:\Tools\PVEs 目录中。

2. 创建 PyTorch 环境¶

2.1. 创建 Python 环境，名称为 PyTorch¶

cd C:\Tools\PVEs
virtualenv PyTorch

2.2. 创建启动脚本¶

创建文件 C:\Tools\PVEs\PyTorch.ps1 ，内容如下，如无法启动请用 PowerShell ISE 编辑器大概编辑一次：

Write-Output "进入 PyTorch 虚拟环境目录，目录地址：C:\Tools\PVEs\PyTorch"
cd C:\Tools\PVEs\PyTorch
.\Scripts\activate

2.3. 安装所需开发包¶

需要安装 CUDA 和 cudnn，这里选择 CUDA 11.6 和 cuDNN 8.4 版本。

CUDA Toolkit Archive | NVIDIA Developer

CUDA 深度神经网络库 (cuDNN) | NVIDIA Developer

cuDNN解压到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6 目录即可。

# 查看驱动
nvcc --version

2.3.2. 安装脚本¶

# 更新环境中的包
python -m pip install --upgrade pip
pip install --upgrade setuptools
pip install --upgrade wheel

# 安装 torch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116


# 安装 JupyterLab
pip install jupyterlab

# 安装 中文语言包，可能提示无法找到满足要求的版本，而无法安装
pip install jupyterlab-language-pack-zh-CN

# 启动 JupyterLab
jupyter-lab

2.3.2. 检验安装¶

python
>>> import troch
>>> torch.cuda.is_available()
cuda是否可用

2.4. 创建项目¶

为了便捷使用该虚拟环境，这里我们使用项目目录的形式来使用该环境。

例如项目目录为 C:\Users\JasperXu\Desktop\TorchStudy

创建文件： 启动环境 - PVE.ps1

# 启动 PyTorch 虚拟环境，打开目录 src
cd C:\Tools\PVEs
.\PyTorch\Scripts\activate
cd C:\Users\JasperXu\Desktop\TorchStudy\src

创建文件： 启动环境 - JupyterLab.ps1

# 启动 JupyterLab，打开目录 src

cd C:\Tools\PVEs
.\TensorFlow\Scripts\activate
cd C:\Users\JasperXu\Desktop\TorchStudy\src
jupyter-lab

创建文件：启动环境 - VSCode.ps1

# 启动 VS Code，打开目录 src
cd C:\Users\JasperXu\Desktop\TorchStudy
code src

创建文件： 启动环境.ps1

# 启动新进程，运行指定脚本
cd C:\Users\JasperXu\Desktop\TorchStudy

Start-Process powershell -ArgumentList "-NoExit","-Command", "& '.\启动环境 - JupyterLab.ps1'"
Start-Process powershell -ArgumentList "-NoExit","-Command", "& '.\启动环境 - PVE.ps1'"
# 学习类的项目直接启动 JupyterLab 就够了，不需要使用 VS Code。
# Start-Process powershell -NoNewWindow -ArgumentList "-Command", "& '.\启动环境 - VSCode.ps1'"

exit

创建日期: 2021-09-20 21:00:00
最后更新: 2022-07-30 02:00:00