Skip to content

GitLab

Unverified Commit 56682040 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[Doc] specify pytorch backend for tutorials

parent 2331feac
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tutorials/blitz/4_link_predict.py
View file @ 56682040
......@@ -18,6 +18,8 @@ By the end of this tutorial you will be able to
"""
import itertools
import os
os.environ['DGLBACKEND'] = 'pytorch'
import numpy as np
import scipy.sparse as sp
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment