2018年10月17日水曜日

Ubuntu16.04 CUDA9.1環境にTorch7を入れる

実行環境は
Ubuntu16.04LTS
CUDA9.1
GTX1080Ti
ドライバのバージョン 390.67

やってみると手順超簡単ですが
調べるのに苦労したのでメモ.

①.bashrcにパスを記入
②install.shオプションTORCH_NVCC_FLAGSをつけて実行.
TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" ./install.sh

以上