VPN: TJU-VPN
IP: ** . ** . ** . **
Server | Port | Manager | HardWare Configuration |
---|---|---|---|
GPU-2080Ti | 101 | 窦明亮 | 48 Thread; 512G Memory; 2080Ti*4; 12*4G GPU Memory; 4T Disk + 4T Disk (protein group) |
GPU-2080Ti | 102 | 李雪健 | 48 Thread; 256G Memory; 2080Ti*4; 12*4G GPU Memory; 4T Disk |
GPU-V100 | 103 | 李佳玮 | 96 Thread; 1024G Memory; V100*4; 32*4G GPU Memory; 4T Disk |
CPU-P700 | 104 | 孟巧珍 | 24 Thread; 384G Memory; 4T Disk |
### 所有人都需在程序中加入以下相应的代码段,特殊情况除外。
### GPU 可用序号:{0, 1, 2, 3}, 使用 nvidia-smi 查看可用的GPU
# 1\ Tensorflow 1.0
import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定要使用的GPU序号
config = tf.ConfigProto()
config.gpu_options.allow_growth=True #不全部占满显存, 动态增长
sess = tf.Session(config=config)
import keras.backend.tensorflow_backend as keras_backend
keras_backend.set_session(sess) #仅对于使用 tensorflow1.0 后端的keras
# 2\ Tensorflow2.0
import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定要使用的GPU序号
gpu_devices = tf.config.experimental.list_physical_devices('GPU')
if gpu_devices:
for gpu in gpu_devices:
tf.config.experimental.set_memory_growth(gpu, True)
# 3\ PyTorch (PyTorch 默认不占用全部显存,只需指定要使用GPU序号即可)
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定要使用的GPU序号