首页 新闻动态 技术文章

Pytorch安装

发布时间:2025-04-11 11:37 点击:80

CUDA 10.1 Python 3.9安装:

pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html


查看CUDA版本:

nvcc -V


查看Python版本:

Python -v


验证:

import torch
print(torch.__version__)