Q&A
前提・実現したいこと
先日，pythonのコードを動かしていたところ，runtime errorが発生しました．
dockerコンテナ内でcuda-9-0,python3.5.9,pytorch1.1.0を用いてGPUを動かしたいのですが，
どのようにすればエラーが回避できるでしょうか．
教えていただきたいです．
発生している問題・エラーメッセージ
RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383
環境
python3.5.9 pytorth1.1.0
nvidia-Driver Version: 410.48
nvcc
1nvcc: NVIDIA (R) Cuda compiler driver 2Copyright (c) 2005-2017 NVIDIA Corporation 3Built on Fri_Sep__1_21:08:03_CDT_2017 4Cuda compilation tools, release 9.0, V9.0.176
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
を実行すると良いといろいろなサイトで書かれていましたが，実行するとpythonが落ちます．
補足情報（FW/ツールのバージョンなど）
使用しているdocker image:nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
あなたの回答
tips
プレビュー
