Arieslong的博客

原创 windows系统下anaconda环境下安装pytorch（CPU+GPU）

直接下载torch 阿里云镜像（快）： pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com。print(torch.cuda.is_available()) #输出为True，则安装成功。查询对应下载版本，有可能版本与python版本不对应。找到python版本对应cuda版本，然后复制下载。有，命令行nvidia-smi检查版本。