-
Notifications
You must be signed in to change notification settings - Fork 50
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.
Already on GitHub? Sign in to your account
Pytorch Install Instructions #515
Comments
Is there a prebuilt version for pytorch >= 1.0? |
@ghostplant yes, see |
@iotamudelta I tried that but torch library is not included in the python runtime: $ docker run -it --rm --network=host --privileged rocm/pytorch:rocm2.9_ubuntu18.04_py3.6 python3 -c 'import torch'
Traceback (most recent call last):
File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'torch' |
Please try the ones with '_pytorch' in their tag, and specify 'python3.6' instead of 'python3' (sometimes the dockers have python3.5 linked to python3). HTH |
@jithunnair-amd Thanks, seems that for rocm-2.9, only py2.7 contains the pytorch but py3.6 doesn't. |
Seems okay on my end:
|
Your both correct, ubuntu18.04 does not have pytorch but ubuntu16.04 does
|
@Delaunay Wiki has been updated for the sed command and dependency update. However, you shouldn't need to modify your LIBRARY_PATH for rccl, as the cmake build should find the rccl library and use the absolute path in the build. Can you please try again and let us know if you still observe the issue with finding rccl library? You can use VERBOSE=1 during the build and capture the log file to observe which build command uses '-lrccl', if any. |
I will close the issue, I do not have access to the machine with AMD GPUs anymore so I wont be able to test again for the RCCL linking issue. |
馃摎 Documentation
Pytorch build instructions are a bit outdated
I also had to add RCCL linking path
export LIBRARY_PATH="/opt/rocm/rccl/lib/"
.Although CMake did say it found it and
ld -lrccl
was not able to find it either.The text was updated successfully, but these errors were encountered: