Pytorch¶
安装¶
官方安装手册 https://pytorch.org/get-started/locally/
打开官方安装手册链接后，网站会检测操作系统和显卡，并推荐下载的版本。
检查 CUDA 版本
如果使用的是 N 卡，可以运行命令如下查看 CUDA 版本。
nvidia-smi
CUDA 版本显示在输出结果的右上角。确保你的 CUDA 版本不低于你选的那个。如果版本过低，你可能要升级显卡驱动。
确认版本无误后，复制表格最下方显示的命令并在激活的环境中运行。
# 这是我的安装命令，你应该去复制网站上的那个而不是下面这个
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu129
注意环境
运行安装命令前请确保你激活了正确的环境。
需下载的总文件大约 4G 左右（因版本而异），时间可能有点长，请确保网络通畅
下载完成后，可通过如下方式测试是否可用
# 运行下行命令以 REPL 模式启动解释器
python
# 解释器启动后，输入下行代码
import torch
# 如果没有报错，那就是正确安装了。
# 如果下载的是 CUDA 的版本，可以输入下行代码检查 CUDA 是否可用
torch.cuda.is_available() # 应该返回 True
torch.cuda.get_device_name(0) # 应该返回显卡名称
# 最后可输入下行代码关闭解释器
exit()