1.安装环境
OS: Windows Server 2016 Standard Evaluation
CUDA: Cuda compilation tools, release 10.1, V10.1.105
Python: Python 3.7.3
conda: conda 4.6.13
因为是在工作站上安装,所以系统是Windows Server的,Windows10的安装方法也一样。
其次,预先安装了Anaconda,所以是Python 3.7 + conda 4.6.13的环境。
建议大家也安装Anaconda,安装简单便捷,反观编译Python3源码实在是太多坑。
2.安装pyorch
确定环境后进入pytorch官网:https://pytorch.org/get-started/locally/
按环境依次选择后,下方会出现一条命令,提示Run this Command.
打开Anacoda Prompt,输入上图红框内的命令:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch