首先需要安装并配置好Anaconda环境,安装教程教程随便找一个就好。
第一步:安装好之后创建conda虚拟环境:
conda create -n your_env_name python=x.x
第二步:安装需要版本的pytorch:pytorch下载链接
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官网