赞
踩
pytorch教程需要torchvision下载数据集,使用pip安装指定版本,首先使用conda list torch查看自己安装torch版本,我的pytorch版本1.9.0对应cuda版本11.1
在以下网址查找对应torchvision版本,https://pytorch.org/get-started/previous-versions/,pytorch1.9.0 torchvision0.10.0
再去一下网站下载对应whl文件,https://download.pytorch.org/whl/torch_stable.html,我的cuda版本是11.1对应cu111,python版本3.8对应cp38,Ubuntu20选择linux_x86,复制链接用wget下载,再使用pip安装
wget https://download.pytorch.org/whl/cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-linux_x86_64.whl
pip install *.whl
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。