-
在安装之前先看一下自己的显卡能不能安装CUDA,如果不能安装或者已经在英伟达官网下载CUDA并安装就跳到第5条。
-
在torch的官网有一键安装pytorch的命令,官网地址: https://pytorch.org/get-started/locally/
-
下面图片中为windows系统,如果有其他操作系统的请自行查看,为了避免不兼容,版本选择10.2。
-
安装torch和CUDA
-
只安装torch
安装torch+cuda笔记
最新推荐文章于 2024-04-23 14:14:48 发布