linux下安装pytorch
时间: 2023-04-24 17:00:09 浏览: 158
Linux安装pytorch教程.docx
可以使用pip进行安装,在终端中输入以下命令:
```
pip install torch torchvision
```
如果需要指定版本安装,可在命令后加上版本号:
```
pip install torch==1.7.0+cu111 torchvision==0.8.2+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
如果需要在GPU环境中运行,还需要安装CUDA和cuDNN。
如果出现权限问题,在命令前加上sudo。
阅读全文