{"cells":[{"metadata":{},"cell_type":"markdown","source":"The main problem is we need to confirm CUDA version and Pytorch CUDA version"},{"metadata":{},"cell_type":"markdown","source":"> ## Setup apex"},{"metadata":{"trusted":true},"cell_type":"code","source":"","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Check CUDA version"},{"metadata":{"trusted":true},"cell_type":"code","source":"!nvcc -V","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"#!pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Check Pytorch CUDA version"},{"metadata":{"trusted":true},"cell_type":"code","source":"#! pip install -v --no-cache-dir --global-option=\"--cpp_ext\" --global-option=\"--cuda_ext\" ../input/nvidiaapex/repository/NVIDIA-apex-39e153a\n#! pip install -v --no-cache-dir --global-option=\"--cpp_ext\" --global-option=\"--cuda_ext\" ../input/nvidiaapex/repository/NVIDIA-apex-39e153a\n#! git clone https://github.com/NVIDIA/apex\n#!pip install -v --no-cache-dir --global-option=\"--cpp_ext\" --global-option=\"--cuda_ext\" ./apex\n#!cd ../input/apex-master/apex-master/apex-master/ && pip install --no-cache-dir --global-option=\"--cpp_ext\" --global-option=\"--cuda_ext\" .\nimport torch\n\nprint(torch.version.cuda)\nprint(torch.__version__)","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Check APEX Lib"},{"metadata":{"trusted":true},"cell_type":"code","source":"import os\nprint(os.listdir(\"../input/nvidiaapex/repository/NVIDIA-apex-39e153a\"))","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Install Apex"},{"metadata":{"trusted":true},"cell_type":"code","source":"!git clone https://github.com/NVIDIA/apex\n!cd apex\n!pip install -v --no-cache-dir   ../input/nvidiaapex/repository/NVIDIA-apex-39e153a","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"import torch\nimport sys\nimport warnings\n\nfrom apex import amp \n\n\n\n\ndevice = torch.device('cuda')\n\n","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"## Done"},{"metadata":{"trusted":true},"cell_type":"code","source":"dir(amp)\n","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"If you have any question, please leave a comment here."},{"metadata":{"trusted":true},"cell_type":"code","source":"","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat":4,"nbformat_minor":4}