search

lanl / OpenFWI

A collection of codes with OpenFWI project
BSD 3-Clause "New" or "Revised" License
61 stars 10 forks source link

Set environment for PyTorch - GPU #2

Closed Abdelhameed88 closed 1 year ago

Abdelhameed88 commented 1 year ago

Hello all,

Thanks for your effort and publish this great code.

Please, I have an issue on setup the environment to run the code for FlatVel-a. I set up the environment using the following commands:

However, I got the following error:

I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.

To enable the following instructions: SSE4.1 SSE4.2 AVX AVX2 AVX512F FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.

Please, any advice.

Thanks in advance, Ahmed

Shihang-LANL commented 1 year ago

Hi, Ahmed

The error you're encountering appears to be related to TensorFlow. However, this specific package does not utilize any TensorFlow functions. It's possible that you have TensorFlow installed in your default path, which can cause compatibility issues with PyTorch. To resolve this, I recommend uninstalling TensorFlow before using PyTorch. Here are the required packages for this code to work properly, the packages other than these may cause imcompatibility: pytorch v1.7.1 torchvision v0.8.2 scikit learn numpy matplotlib (for visualization)

Best regards, Shihang

Abdelhameed88 commented 1 year ago

Many thanks Shihang for your suggestions. It works now.

Best regards, Ahmed