下载对应版本的torch+cuda
时间: 2024-01-02 19:02:04 浏览: 89
cuda9.0对应需要的torch&torchvision
根据提供的引用内容,如果您想下载对应版本的torch cuda,可以使用以下命令安装:
pip install torch==1.7.0 cu101 torchvision==0.6.1 cu101 -f https://download.pytorch.org/whl/torch_stable.html
阅读全文
cuda9.0对应需要的torch&torchvision