Install PyTorch 1.7 with CUDA 10.1 on MacOS High Sierra 10.13.6 + GTX 1060 6GB

pytorch-with-cuda-on-macos.md

Install PyTorch 1.7 with CUDA 10.1 on MacOS High Sierra 10.13.6 + GTX 1060 6GB

Install web driver for NVIDIA

1. install webdriver.sh

brew tap vulgo/repo
brew install webdriver.sh

2. install lastest driver

webdriver

installed version 387.10.10.10.40.133

source

Install CUDA

1. install cuda via brew:

brew tap homebrew/cask-drivers
brew cask install nvidia-cuda

2. add the following to your file ~/.bash_profile:

export PATH=/Developer/NVIDIA/CUDA-9.0/bin${PATH:+:${PATH}}
export DYLD_LIBRARY_PATH=/Developer/NVIDIA/CUDA-9.0/lib${DYLD_LIBRARY_PATH:+:${DYLD_LIBRARY_PATH}}

3. verify installation

nvcc --version

installed version 418.105

source

Install cuDNN

1. download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1
2. untar tar -xzvf cudnn-10.1-osx-x64-v7.6.5.32.tgz
3. copy the files into the CUDA Toolkit directory, and change the file permissions

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib/libcudnn* /usr/local/cuda/lib
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

installed version 7.6.5

source

Build PyTorch from source

1. install anaconda
2. remove previous versions of pythorch

conda remove pytorch torchvision -y
pip uninstall torch -y
pip uninstall torch -y  # yes twice

3. install dependencies

conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests

4. get pytorch source

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch

5. install pytorch

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
MACOSX_DEPLOYMENT_TARGET=10.13 CC=clang CXX=clang++ python setup.py install

error when linking libtorch_cuda.dylib on macOS 10.13.6 fix

6. build torchvision from source

git clone --recursive https://github.com/pytorch/vision
cd vision
export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
MACOSX_DEPLOYMENT_TARGET=10.13 CC=clang CXX=clang++ python setup.py install

7. verify installation

import torch
print(torch.__version__)

import torch
assert torch.cuda.is_available()
assert torch.randn(20).cuda().is_cuda

import torch
assert torch.backends.cudnn.enabled

installed version 1.7.0a0+eeb43ff

@brokeyourbike, what versions of the dependencies and what version of Python did you use? Can you capture your exact environment in pip or conda? Can you share the result of conda list --explicit?

Hi @geoffwoollard I have performed this setup more then a year ago and now I don't have it installed anymore. A lot of things chnanged in my setup over the years. Unfortunately I cannot share the result of conda list --explicit

Do you still have the pytorch/build directory? It has informative log files like pytorch/build/CMakeFiles/CMakeError.log and pytorch/build/CMakeFiles/CMakeOutput.log

Also, I'm getting different errors depending on what PyTorch version (commit) I try to install, and what version of Python (3.7, ..., 3.10) I use.

@geoffwoollard nah, sorry, don't have PyTorch installed right now, latest macOS versions don't support Nvidia GPU