未验证 提交 92d14840 编写于 作者: X xiaoxiaohehe001 提交者: GitHub

nvcc_centos (#44832)

上级 bc106fad
...@@ -16,9 +16,11 @@ set -e ...@@ -16,9 +16,11 @@ set -e
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY ## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY
# set cicc PATH for Centos
export PATH=$PATH:/usr/local/cuda/nvvm/bin
# check nvcc version, if nvcc >= 11.7, just run nvcc itself # check nvcc version, if nvcc >= 11.7, just run nvcc itself
CUDA_VERSION=$(nvcc --version | grep -oP '(?<=V)\d*\.\d*') CUDA_VERSION=$(nvcc --version | grep -oP '(?<=V)\d*\.\d*')
CUDA_VERSION_MAJOR=${CUDA_VERSION%.*} CUDA_VERSION_MAJOR=${CUDA_VERSION%.*}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册