总结:
首先使用nvcc -V查看cudatoolkit的版本:
(ENVv) acat@acat-xx:paper_or_gcn$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89
要确保安装的cuda版本大于等于10.1.不然的话就会失败。
pytorch-fast-transformer简介: