We’re still establishing the best practices. There are two main choices I see:
- Use Lambda Stack as an easy way to install your Drivers, CUDA, CuDNN, etc.
- Use Lambda Stack’s version of TensorFlow / PyTorch.
If you decide to go for type 1, you’ll simply install Lambda Stack and then create your virtualenv like this:
virtualenv -p python3 your_venv
If you decide to go for type 2, you’ll install Lambda Stack and then create your virtualenv like this:
virtualenv --system-site-packages -p python3 your_venv
Essentially in type 2 you’ll be using our TensorFlow / PyTorch packages and type 1 you’ll be installing them yourself.