Adrianna的专栏

在python文件第一行加上以下语句，这个python脚本就只会使用id为“2”的gpu。import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "2"# use multiple gpus# os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"...

reference:https://stackabuse.com/image-classification-with-transfer-learning-and-pytorch/#visualizationhttps://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html#writing-to-tensorboardh...

pytorch提供了一个很方便的类torchvision.datasets.ImageFolder用于读取文件夹中的图片数据。我们只要把文件夹中的图片文件按照以下规则存放：root/dog/xxx.pngroot/dog/xxy.pngroot/dog/xxz.pngroot/cat/123.pngroot/cat/nsdf3.pngroot/cat/asd932_.png通过Ima...

本地已经安装了anaconda3+cuda9.0+tensorflow1.7，打算转pytorch，因此记录下pytorch的环境安装。cuda和anaconda的安装可以参考之前的博文：点我直达1. 新建anaconda环境新建一个pytorch的环境，python版本建议选择3.6conda create -n pytorch python=3.62. 安装pytorch因为本地...