-
创建环境conda create -n pytorch python=3.7
-
激活环境conda activate pytorch
-
进入 官网PyTorch.,下拉网页:
会自动根据电脑显示你可安装的CUDA版本 -
最后输入官网提供的命令conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch,即可下载
-
测试代码:
import torch
torch.cuda.is_available()
pytorch安装
最新推荐文章于 2024-07-18 20:20:15 发布