首先，查看显卡驱动版本：
nvidia-smi
image.png
安装torch：
地址：https://pytorch.org/get-started/locally/
可使用在线安装，按需选择，然后生成脚本，比如：
image.png
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
或者下载whl后安装(https://download.pytorch.org/whl/torch/)。
按自己的显卡需要，尽可能安装最新的（有时间显卡并没有最新的驱动，只需要装一个能支持的、最新的就好）。
注意，要选择带 +cu的，就是这样的：torch-2.1.1+cu121-cp310-cp310-win_amd64.whl
背景提要：刚刚安装了pytorch，torch-geometric、torch_scatter和torch_spa...
安装Torch时的一些心得 如果你想安装GPU版本的，在安装Torch之前，最好先安装并配置好Cuda和Cudnn...
背景： 在安装torch的时候，pip install torch 然后安装失败，告诉我需要更新我的pip，于是我...
由于stackgan项目需要，来安装个torch http://torch.ch/docs/getting-sta...
1.cuda 以及cudnn的安装文档 2.torch1教程 3.torch2教程
成功安装，但pycharm依然不能导入torch
我在MacOS下安装torch一直不成功，所以用虚拟机运行Linux，在Linux下安装了Torch以及PyTor...
背景：安装torch_scatter和torch_sparse（不是pytorch）的过程中，发现了错误。 错误提...
参考： https://blog.csdn.net/dongfang1984/article/details/81...
1、在终端运行以下3句命令，会将Torch安装在主目录下，如图所示： 2、刷新环境变量配置 3、添加环境变量 i:...
本文标题：torch 安装
本文链接：https://www.haomeiwen.com/subject/ojhnwdtx.html
网友评论