Windows本地配置带GPU的Pytorch环境

首先需要安装并配置好Anaconda环境,安装教程教程随便找一个就好。

第一步:安装好之后创建conda虚拟环境:

conda create -n your_env_name python=x.x 

第二步:安装需要版本的pytorch:pytorch下载链接
Windows本地配置带GPU的Pytorch环境_第1张图片
cu100/torch-1.2.0-cp36-cp36m-win.amd64.whl

  • cu100:表示cuda版本为10.0,
  • torch-1.2.0:表示torch版本为1.2.0
  • cp36:表示适用python版本为3.6
  • win:表示适用windows系统
  • x86_64:表示同时兼容32和64位系统

第三步:安装对应的CUDA版本:CUDA官网
Windows本地配置带GPU的Pytorch环境_第2张图片
Windows本地配置带GPU的Pytorch环境_第3张图片

你可能感兴趣的:(windows,pytorch,人工智能)