依次输入以下指令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

conda install pytorch torchvision cudatoolkit=10.0

验证GPU可用:

>>import torch

>>print(torch.cuda.is_available())

True

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄