Advertisement

PyTorch使用的早期问题

阅读量:

一、GPU支持:

第一步访问PyTorch的主页面,并点击“安装”按钮

如上图所示,可以直接运行其中的

复制代码

命令就可以安装好PyTorch对GPU的支持。当然还要安装相应的CUDA驱动。

可以参考我的另外一个对TensorFlow的内容[conda 安装CUDA 和cudnn_bjay的博客-博客_conda安装cuda]( "conda 安装CUDA 和cudnn_bjay的博客-博客_conda安装cuda")

当尝试从Python Imaging Library (PIL)模块中导入名为Pillow_VERSION的名称时会引发 ImportError:无法从PIL模块中导入名为Pillow_VERSION的名称,并且堆栈跟踪显示错误发生于D:\program\Anaconda3\lib\site-packages\PIL_init_.py位置。

在执行如:python main.py -a resnet50 --lr=0.015 --batch_size=128 --dist_url=

全部评论 (0)

还没有任何评论哟~