benchmark
2fcedc4b - Update conda installation command to handle the installation difference with cudatoolkit and pytorch-cuda (#1260)

Commit
3 years ago
Update conda installation command to handle the installation difference with cudatoolkit and pytorch-cuda (#1260) Summary: This PR fixes the conda installation issue with pytorch 1.12.1. It was observed that "conda install --force-reinstall pytorch=1.12.1 torchvision torchtext pytorch-cuda=11.6 -c pytorch -c nvidia" command installs pytorch 1.12.1 but with cuda 11.3 instead of 11.6. Pull Request resolved: https://github.com/pytorch/benchmark/pull/1260 Reviewed By: xuzhao9 Differential Revision: D40911616 Pulled By: weiwangmeta fbshipit-source-id: 3cfa67be7420baf9b90dc31d5fa7ecab3e97d053
Author
Parents
Loading