pytorch,torch,torchvision的gpu版本安装避坑
在线or离线安装
最近使用pytorch跑深度学习模型,需要用到与gpu相对应的包,写此博文,提醒后来者避避坑
pytorch
本次使用python3.6(conda环境),一般conda环境可以创建多个Python版本,互不干扰,比单一Python要好用,在pycharm上添加pytorch库。
点击左下方 + ,进去页面输入pytorch,安装对应的版本
torch和torchvision
torch,torchvision库在pycharm中没搜到,在conda环境下激活python36,使用pip命令安装
在使用计算机的狗之前,要安装好显卡驱动,cuda和处cudnn,请至
https://blog.csdn.net/cashmood/article/details/105081586
因为要是用gpu跑代码,所以和torch,torchvision相关的包都必须是和gpu相对应的,
(1)离线安装(强烈推荐)
下载对应的torch和torchvision :http://download.pytorch.org/whl/torch_stable.html
或者:(第三方扩展包&#x