摘要:
import osos.environ['CUDA_VISIBLE_DEVICES'] = '0, 1, 2'import torch #注意以上两行先后顺序不可弄错 device = torch.device('cuda') model = DataParallel(model)model.to( 阅读全文
摘要:
`sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb` `sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1 阅读全文