欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Linux安装pytorch

程序员文章站 2022-06-11 22:13:57
...

linux安装pytorch

pytorch官网
pytorch各种旧版本

#CUDA 9.2

pip install torch==1.7.1+cu92 torchvision==0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

方法:打开anaconda里你创建的虚拟环境,将上面的命令进去等待下载即可。
注:这里是GPU版本torch安装