win10搭建pytorch-gpu环境
一 准备条件
- 首先需要在电脑上安装anaconda
具体安装过程可参考:Anaconda详细安装及使用教程(带图文) - 查看电脑支持的CUDA版本。
NVDIA控制面板–>帮助–>系统信息–>组件
电脑的cuda版本需要和安装pytorch时下载的cuda版本相同。
二 在anaconda中新建环境
可以先使用conda
命令查看conda是否安装好
接着使用命令行方式创建一个新的环境:
conda create -n pytorch python=3.6.3 anaconda
其中pytorch为创建的环境名称,python=3.6.3为安装的