-Mr.A- 現在書くことが思いつかないため更新停滞中
ページ
ホーム
イラスト
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
以上
前の投稿
ホーム