安装torch+cuda笔记

2021/11/6 23:41:58

本文主要是介绍安装torch+cuda笔记,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

  1. 在安装之前先看一下自己的显卡能不能安装CUDA,如果不能安装或者已经在英伟达官网下载CUDA并安装就跳到第5条

  2. 在torch的官网有一键安装pytorch的命令,官网地址: https://pytorch.org/get-started/locally/

  3. 下面图片中为windows系统,如果有其他操作系统的请自行查看,为了避免不兼容,版本选择10.2。

  4. 安装torch和CUDA在这里插入图片描述

  5. 只安装torch在这里插入图片描述



这篇关于安装torch+cuda笔记的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程