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