安装 torch

安装 torchpytorch 的安装 安装 torch

大家好,欢迎来到IT知识分享网。

torch 的安装分为两个版本:Cpu 和 Gpu

CPU 版本 torch 的安装:(安装 anacode 的前提下)

1、控制台输入:

pip install torch==1.8.0

2、安装 torchvision

pip install torchvision 

GPU版本torch 的安装:

1、查询 GPU 所对应的CUDA 版本号

nvidia -smi

2、CUDA 官网下载(NVIDA.DEVELOPER),推荐下载 11.3 的版本(经典!)

cp3.6=python 3.6

cu101=cuda 10.1

3、下载torch到本地

4、下载 torchvision

另:如果已经安装 GPU 版本的 torch,通过命令查询

ncvv -V

免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://haidsoft.com/122825.html

(0)
上一篇 2025-10-13 19:45
下一篇 2025-10-13 20:10

相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注微信