大家好,欢迎来到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