其他分享
首页 > 其他分享> > win安装pytorch

win安装pytorch

作者:互联网

cpu版:
在https://pytorch.org/get-started/locally/找对应下载信息,命令行输入pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple/
GPU版:看控制面板有没有nvidia,如果有就说明是GPU版,管理3D设置->组件->找到cuda版本,再在https://pytorch.org/get-started/locally/找对应下载信息,命令行输入conda install pytorch torchvision cudatoolkit =10.1 ,下载好在命令行输入:python->import torch如果不报错,就说明安装成功.

标签:locally,https,win,torch,pytorch,命令行,安装,torchvision
来源: https://blog.csdn.net/pwgmlyrics/article/details/121311217