设置GPU可见

export CUDA_VISIBLE_DEVICES=1 Only device1 will be seen

export CUDA_VISIBLE_DEVICES=0,1 Devices 0 and 1 will be visible

export CUDA_VISIBLE_DEVICES=0,2,3 Devices 0,2,3 will be visible;device 1 is masked


通常可以将这行加入用户home目录下.bashrc,这样每次远程登录就不用重新配置了。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容