Skip to content

gpu support from pip install version #2

@szrlee

Description

@szrlee

When I use pip to install scs and test with solve_random_cone_prob_gpu.py

~/scs-python$ python test/solve_random_cone_prob_gpu.py 
Traceback (most recent call last):
  File "test/solve_random_cone_prob_gpu.py", line 75, in <module>
    main()
  File "test/solve_random_cone_prob_gpu.py", line 14, in main
    solve_feasible()
  File "test/solve_random_cone_prob_gpu.py", line 34, in solve_feasible
    sol_i = scs.solve(data, K, gpu=True, **params)
  File "/home/xxxx/anaconda3/lib/python3.6/site-packages/scs/__init__.py", line 56, in solve
    import _scs_gpu

Must I install scs-python from source if I want the GPU feature?
Another question, if possible to be answered, is that when I use GPU version pytorch or tensorflow combining with this GPU-version scs, will there be some additional data transferring costs? I suppose not since the data are already loaded to GPU.

Thanks a lot for attention!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions