当前位置:网站首页>Pytorch CPU/GPU 安装方法。
Pytorch CPU/GPU 安装方法。
2022-06-10 03:47:00 【Vertira】
首先,查看自己有没有显卡及其显卡驱动。没有显卡,就CPU安装就行。
windows版,CUDA,cudnn,和Anaconda安装方法这里不详述。参考一下连接
如果你没有GPU,只需安装Anaconda3 就行了,CPU也可以使用。
这里只介绍在你安装成功以上之后,进行Pytorch安装
1、进入pytorch官网
PyTorch
https://pytorch.org/CPU安装
看好版本,然后最后一行,Run this Command:
conda install pytorch torchvision torchaudio cpuonly -c pytorch运行这个就行了。
如果你的是GPU ,你就选择对于的CUDA版本,然后运行对于的Run this Command
GPU安装

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch注意:下载速度慢的解决方案如下
下载速度慢或者下载老是中断的小伙伴,可以考虑使用镜像源下载,常用的镜像源是清华镜像源。在粘贴Run this Command后面的代码之前可以添加如下几行语句:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/这样下载速度就快很多
边栏推荐
- Redis core technology and practice - practice reading notes 20 ~ end
- Yolov5 target detection neural network -- calculation principle of loss function
- Understanding of analog-to-digital conversion
- vulnhub之hacksudo:Thor
- c语言刷题系列(二)
- [MySQL] 数据库-视图
- [paper notes | deep reading] struc2vec: learning node representations from structural identity
- [9 techniques for accelerating pytorch model training]
- [yolov5.yaml analysis]
- Distributed current limiting: current limiting based on sentinel implementation (I) - Overview
猜你喜欢

Yolov5 target detection neural network -- calculation principle of loss function

Dapr - what are the advantages of the microservice framework used by major manufacturers?

在线文本字符串批量替换工具

分布式事务解决方案

【TFLite, ONNX, CoreML, TensorRT Export】

Opencv_ 100 questions_ Chapter I (1-5)

All MySQL collections from 0 to 1 are highly recommended

New function official publicity playfab makes cross platform game architecture so simple

Redis core technology and practice - practice reading notes 20 ~ end

document editor
随机推荐
Implementation scheme of shared file
【pytorch 修改预训练模型:实测加载预训练模型与模型随机初始化差别不大】
在线文本字符串批量替换工具
[mysql] database - View
[mainstream nivida graphics card deep learning / intensive learning /ai computing power summary]
Opencv_ 100 questions_ Chapter I (1-5)
OpenAI新博客:训练大模型的技术
[pytorch pre training model modification, addition and deletion of specific layers]
Business card wechat applet error version 2
SSTI(模板注入) ——(8)
Code error writing method of wechat applet rotation chart
【 YOLOv3中Loss部分计算】
Brush questions_ Linked list related series
Decision engine system & real-time index calculation & risk situation awareness system & risk data list system & fraud intelligence system
C encapsulates fluentvalidation. After using it, the whole article is still abstractvalidator. I really can't see it anymore
【yolov3损失函数】
【PyTorch 模型剪枝实例教程2(结构化剪枝)】
Twitter agreed to open the database for musk to check
RPC practice and core principles - Advanced notes
先序遍历二叉树