banner
xingli

xingli

猫娘爱好者

安装torch1.7

安装 torch1.7#

1. 安装好 cuda 和 cudnn#

具体可看我之前发布的文章

2. 在官网找到适合自己的版本#

Previous PyTorch Versions | PyTorch
image
我安装的是 cuda11 windows 系统 使用这个命令 不同的系统在这里的安装命令都不太一样,具体看官方的代码

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

image
没有报错就是安装完毕了,如果安装很慢请换 pip 的源

加载中...
此文章数据所有权由区块链加密技术和智能合约保障仅归创作者所有。