How to install pytorch 0.4.1 and cuda 9.1, cudnn 7.0 on docker?

1

Hello all, I have success install pytorch using conda command. Now, I want install pytorch 0.4.1, cuda 9.1 and cudnn 7.0 on ubuntu 14.04 and docker? Anyone has success to install please give me some advice? Thanks

2

Try Checking this Docker file

https://hub.docker.com/r/floydhub/pytorch/tags/![02%20AM|690x160](upload://mVTmO8wcVnfGdUQy4TwcXZYqu4F.png)

You can upgrade some if you need

3

Thanks but it only for 0.4.0 and i need newest version 0.4.1

4

yeah after pulling the docker just update the version to 0.41. [Hoping if you don’t have any update constraints]