torch2.4.0安装

1、新建一个虚拟环境

conda create -n torch2.4.0(环境名称) python=3.9(指定python版本，不低于3.8)

conda create -n torch2.4.0 python=3.9

2、激活虚拟环境

conda activate torch2.4.0

记得改成自己的环境名称！！！

3、选择自己的配置，从官网复制代码

（我自己的是windows11，cuda12.1，选择通过pip安装）

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

从官网下载会比较慢

可以选择国内镜像源清华大学、阿里云、豆瓣、中国科技大学等

（但是可能会出现一些小问题，看自己需求）

4、安装完成

5、将虚拟环境添加为jupyter内核

pip install ipykernel

python -m ipykernel install --user --name=torch2.4.0

查看内核列表

jupyter kernelspec list

已添加完成。

6、在jupyter notebook内检查，切换内核为torch的内核或者以torch的内核新建ipynb文档。

运行下面代码：

import torch

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)

输出结果如上表示torch安装完成。