Hacker News new | ask | show | jobs
by eghri 3134 days ago
If you want to use the p3 instances with Tesla V100's, you'll need to use CUDA9 instead of CUDA8, and unfortunately none of Amazon's AMI's have the latest PyTorch that support CUDA9. The easiest way to use them is get one of the Conda Deep Learning AMI's and then follow the latest PyTorch install instructions, which seems to work pretty seamlessly.