conda create --name torch040 -y python=3.6 pytorch=0.4.0 torchvision cuda90 cudatoolkit=9.0 tensorboardX -c pytorch

conda activate torch040