跳转至

PyTorch 环境搭建

1. 安装和配置 Python

1.1. 安装 Python

下载页面 下载对应的版本。默认下载为 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
:: @cmd /k

2.3. 安装所需开发包

需要安装 CUDA 和 cudnn

# 查看驱动
nvcc --version
2.3.2. 安装脚本
python -m pip install --upgrade pip
pip install --upgrade setuptools
pip install --upgrade wheel

pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio===0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
pip install notebook
2.3.2. 检验安装
python
>>> import troch
>>> torch.cuda.is_available()
cuda是否可用

2.4. 创建项目

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

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

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

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

创建文件: 启动环境 - Jupyter Notebook.ps1

# 启动 Jupyter Notebook,打开目录 src

cd C:\Tools\PVEs
.\TensorFlow\Scripts\activate
cd C:\Users\JasperXu\Desktop\MyProjectA\src
$env:LANG="zh_CN.UTF-8"
jupyter notebook

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

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

创建文件: 启动环境.ps1

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

Start-Process pwsh -ArgumentList "-NoExit","-Command", "& '.\启动环境 - Jupyter Notebook.ps1'"
Start-Process pwsh -ArgumentList "-NoExit","-Command", "& '.\启动环境 - PVE.ps1'"
Start-Process pwsh -NoNewWindow -ArgumentList "-Command", "& '.\启动环境 - VSCode.ps1'"

exit

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

评论