当前位置：网站首页>Pytorch, specify the problem of the graphics card

Pytorch, specify the problem of the graphics card

2022-07-22 21:33:00 【Lost in code】

Run on the server PyTorch Training code for , It turned out that there was no problem . But reinstall cuda after , Cannot specify which blocks GPU Run on , Only in order from 0 Block start using .
 

resolvent ： stay os Followed by the designated graphics card .（ Because when importing torch Behind the library , It has called the graphics card .）

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "4,5,6,7"
 
import torch

 

 

Game Over!

原网站

版权声明
本文为[Lost in code]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/203/202207221105143530.html

边栏推荐

猜你喜欢

随机推荐