本文共 2227 字,大约阅读时间需要 7 分钟。
这篇文档详细提供了在 Windows 10 系统上搭建 CUDA 10.0、PyTorch 1.2.0 和 TensorVision 0.4.0 开发环境的步骤说明。对于深度学习涉及图形处理任务的开发者来说,这是一个常用的配置,以下内容将引导你顺利完成环境搭建。
在开始配置前,请确保你的系统满足以下条件:
PyTorch 和相关工具的安装需要使用 Python 环境。在这里,推荐使用 Anaconda 可容东西包管理器,这样可以方便地管理开发环境。
访问 Anaconda 官方网站:前往 Anaconda 官网 下载安装文件。
下载并安装 Anaconda:
设置 Anaconda 环境: -安装完成后,打开 Anaconda 存储库DLL 文件,打开 Command Prompt,输入 conda create -n pytorch anaconda-python 3.8。
激活自定义环境:
conda activate pytorch,打开 Anaconda 环境。CUDA 是图形处理的加速库,用于 PyTorch 的加速。在此版本中,我们会安装 CUDA 10.0 和相应的 cuDNN。
访问 NVIDIA 官方下载页面:前往 CUDA 10.0 下载页面。
选择合适的 CUDA 版本:
安装 CUDA:
安装 cuDNN:
配置 cuDNN 环境:
nvcc -V,确认 CUDA 版本是否正确安装。访问 PyTorch 官方下载页面:前往 PyTorch 官网下载页面。
下载 PyTorch 和 TensorVision:
.whl 文件(例如 torch-1.2.0-cp36-cp36m-win_amd64.whl 和 torchvision-0.4.0-cp36-cp36m-win_amd64.whl),确保版本与 CUDA 和 cuDNN 一致。安装依赖项:使用 pip 命令安装:
pip install torch-1.2.0-cp36-cp36m-win_amd64.whlpip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
在已有环境中打开命令提示符。
根据需要输入以下命令,确保下载路径正确:
pip install torch-1.2.0-cp36-cp36m-win_amd64.whlpip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
验证 PyTorch 版本:
import torchprint(torch.__version__)
预期输出:torch 1.2.0
验证 CUDA 加速:
import torchx = torch.randn(3, 3)print(x_folder.device) # 检查设备是否为 CUDA
环境变量路径:
PATH 环境变量中添加了 CUDA 的安装目录。environment variables,手动添加路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\10.0\bin。CuDNN 版本匹配:
系统重启:
通过以上步骤,你已经成功搭建了 CUDA 10.0、PyTorch 1.2.0 和 TensorVision 0.4.0 的开发环境。如果在安装过程中遇到问题,可以参考 PyTorch 官方文档 进行进一步的求助。
转载地址:http://qlhtz.baihongyu.com/