pycharm安装pytorch

在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命令行里,运行即可。

之后等待十分钟或者半小时左右,即可安装完毕。

猜你喜欢

转载自blog.csdn.net/sinat_38079265/article/details/121164671