Skip to content

GitLab

Commit f9dc8b15 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Remove CUDA_VISIBLE_DEVICES setting

parent 96809433
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/data/mmcif_parsing.py
View file @ f9dc8b15
...@@ -476,7 +476,7 @@ def get_atom_coords( ...@@ -476,7 +476,7 @@ def get_atom_coords(
all_atom_mask[res_index] = mask all_atom_mask[res_index] = mask
if zero_center: if zero_center:
binary_mask = all_atom_mask.astype(np.bool) binary_mask = all_atom_mask.astype(bool)
translation_vec = all_atom_positions[binary_mask].mean(axis=0) translation_vec = all_atom_positions[binary_mask].mean(axis=0)
all_atom_positions[binary_mask] -= translation_vec all_atom_positions[binary_mask] -= translation_vec
......
train_openfold.py
View file @ f9dc8b15
...@@ -2,7 +2,7 @@ import argparse ...@@ -2,7 +2,7 @@ import argparse
import logging import logging
import os import os
os.environ["CUDA_VISIBLE_DEVICES"] = "4,5" #os.environ["CUDA_VISIBLE_DEVICES"] = "0"
#os.environ["MASTER_ADDR"]="10.119.81.14" #os.environ["MASTER_ADDR"]="10.119.81.14"
#os.environ["MASTER_PORT"]="42069" #os.environ["MASTER_PORT"]="42069"
#os.environ["NODE_RANK"]="0" #os.environ["NODE_RANK"]="0"
......
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