当前位置:   article > 正文

Pytorch安装教程(Python)_python安装pytorch教程

python安装pytorch教程
本教程首先要安装并配置anaconda,如果不具备此条件,请先安装anaconda。
参考链接:Anaconda安装配置教程(Python)

1、查看本机的CUDA版本

cmd命令行输入nvidia-smi,在第一行最右边可以看到CUDA的版本号。

 2、安装Pytroch

1、点击进入Pytorch官网

 2、下拉至下图位置,只需保证系统和电脑匹配,CUDA版本小于本机CUDA版本,调整结束后,复制红框中的指令

 3、打开Anaconda Prompt,粘贴指令,回车执行

 4、输入y确认,回车

5、等待安装直到左上角出现done

6、输入pip list,查看安装了哪些包

 

7、输入python进入环境,输入import torch,无报错代表导入成功(这里要等待一段时间)

 8、输入torch.cuda.is_available()查看torch是否可以使用显卡,True代表可以

9、正常情况到这里,pytorch的环境已经安装好了,如果发现Pycharm中仍报错找不到包,说明是 pycharm中的解释器没有改成anaconda中的python.exe。

解决办法参考:http://t.csdn.cn/T1ptC

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/371648
推荐阅读
相关标签
  

闽ICP备14008679号