InnovArul/torch_extn.sh

Last active October 25, 2021 21:27

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/InnovArul/3eb739d5cb1ebf95c0963afca64e8b35.js"></script>
Save InnovArul/3eb739d5cb1ebf95c0963afca64e8b35 to your computer and use it in GitHub Desktop.

Download ZIP

to build torch extension for all cuda architectures

Raw

torch_extn.sh

	TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0 7.5 8.0 8.6+PTX" python setup.py build
	python setup.py install

	D:\cmder_mini\Cmder.exe "%ActivDir%"

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment