Setting up Unity Machine Learning Agents Toolkit on Windows 10

Recently I wanted to evaluate Unity ML-Agents Toolkit on my PC running windows 10 however it ended up being troublesome setting everything up accurately. The guides I found were mostly out of date because CUDA Toolkit has evolved with various updates and made the instructions obsolete.

If you like to install Unity ML-Agents Toolkit you can follow these instructions which will set you up with the latest versions of the different parts.

Requirements and Software Versions

CUDA compatible graphics card

CUDA Toolkit 10.1 Update 2

cuDNN v7.6.5 (November 5th, 2019), for CUDA Toolkit 10.1

Unity3D 2019.3 or above

Download

Download Unity Hub

Download Anaconda

Download ml-agents release1 repo

Download CUDA Toolkit 10.1 update2 Archive

Download cuDNN

Install

Install Unity

First install Unity Hub, then go to license management in Unity Hub and activate a personal license.