在pycharm上安装pytorch时候,踩了不少坑,终于安装好了。下面是教程:
1.安装前的准备工作
- 最好先将conda或者pip的镜像源更换为国内的镜像,否则会报莫名其妙的超时错误或者失败等等。
- 并且将python版本升级到3.6.2以上,否则python版本过低也会报错,这里可以使用conda创建虚拟环境,选择你想要的python版本。
2.确定你的cuda版本
cmd命令行查看cuda版本
nvidia-smi
3.命令行的方式安装
去pytorch官网:Start Locally | PyTorch
选择stable稳定版本的即可,os选择window,语言选择python,cuda选择跟你驱动一样的版本。
当然你也可以选择CPU版本的,只是一般都会跑在GPU上。
可以选择使用pip或者conda安装,这个看个人喜好。
将相应的pip或者conda安装的命令,粘贴在pycharm命令行里,运行即可。
之后等待十分钟或者半小时左右,即可安装完毕。