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
最后更新: 2022-07-30 02:00:00