最近人工智能等多门课需要复现论文,近两年的论文很多都是基于Pytorch环境做的实验,所以,这里总结一下Pytorch的安装教程,做好最快、最简单、最好地完成安装。
本机环境 |
---|
Win10+1050Ti+Python3.7 |
1、查看本机的CUDA版本
cmd命令行输入nvidia-smi
,在第一行最右边可以看到CUDA的版本号,我的版本是11.1
2、安装Pytroch
然后选择Get Started,就是如下界面
2、这里进行Pytorch版本的选择,首先我选择的是Stable稳定版,然后OS是Windows系统,Package包就使用Conda,Language肯定选Python,最后的Compute Platform就根据大家的需求来定了。大家如果想在自己电脑(具有NVIDIA显卡)上用GPU进行深度学习模型的推理,就选CUDA;如果只想调用CPU运行算法模型的推理,就选CPU。