cuda10.2安装pytorch1.8

CUDA 10.2环境下正确安装 PyTorch 1.8 的方法
为了在 CUDA 10.2 环境下正确安装 PyTorch 1.8，可以按照以下方式操作：
创建 Conda 虚拟环境
首先通过 conda 创建一个新的 Python 3.7 虚拟环境并激活它：
conda create --name pytorch_env python=3.7
conda activate pytorch_env
登录后复制
这一步是为了确保有一个干净的环境来安装所需的依赖项[^1]。
安装 PyTorch 和相关库
在虚拟环境中执行以下命令以安装支持 CUDA 10.2 的 PyTorch 1.8 及其配套库：
pip install torch==1.8.0+cu102 torchvision==0.9.0+cu102 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
登录后复制
上述命令会从官方指定的 URL 中下载适合 CUDA 10.2 的版本[^3]。注意 -f 参数的作用是指定额外的索引链接地址，从而获取特定于 CUDA 版本的二进制文件。
验证安装是否成功
完成安装后可以通过运行以下脚本来验证 PyTorch 是否能够正常调用 GPU：
import torch
print(torch.cuda.is_available()) # 如果返回 True，则说明 GPU 已启用
print(torch.__version__) # 打印当前使用的 PyTorch 版本号
print(torch.version.cuda) # 显示所连接的 CUDA 版本
登录后复制
如果以上测试均显示预期的结果，则表明安装过程顺利完成[^2]。