@SimonW, @smth

  
      gist.github.com
  
  
    https://gist.github.com/dakshanand/34b9b1daadb8dd74f470b84fbaf70dd2 162
Agents.py
import math
import copy
import time
import random
import threading
import matplotlib
import numpy as np
from envHandler import Env

import torch
This file has been truncated. show original




  
  
    
    
  
  


Line 21 - I am using torch.multiprocessing
Line 240 - I am using ‘spawn’ method
Line 162 - torch.cuda.set_device(1)  [ If I put 0 here it works fine]