最新推荐文章于 2024-10-20 15:51:23 发布
xyhhhhhhhha 于 2020-10-28 16:32:33 发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
step1:确认自己的cuda的版本号
nvcc --version 1
例如我的，显示结果如下：
由结果可以看出cuda版本号是9.0
step2:去官网查看命令
https://pytorch.org/
注意：如果cuda 不匹配，命令是无法使用的，比如我的cuda是9.0，则命令是：
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch 1
step3：测试是否安装成功：
(pytorch-3) XXXX>python Python 3.5.6 |Anaconda, Inc.| (default, Aug 26 2018, 16:05:27) [MSC v.1900 64 bit (AMD64)] on win32 Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> import torchvision 123456
如import不报错，则安装成功
相关知识
pytorch python=3.6环境安装
创建虚拟环境并,创建pytorch 1.3.1
Win11+Ubuntu20.04双系统安装教程（避坑版）
记tensorflow1.10.0版本安装
基于卷积神经网络的棉花病虫害识别与防治系统，resnet50，mobilenet模型【pytorch框架+python源码】
Pytorch入门——手把手教你MNIST手写数字识别
【深度学习目标检测】二十五、基于深度学习的花卉分类系统
基于Python和PyTorch的小程序苹果病虫害识别教程
Python基于Pytorch Transformer实现对iris鸢尾花的分类预测，分别使用CPU和GPU训练
PyTorch 深度学习实战
网址: anaconda安装pytorch GPU版小白教程 https://www.huajiangbk.com/newsview1163528.html
| 上一篇: 机器学习教程 之 SKlearn |
下一篇: GTA5辅助Bob教程 |
