- 博客(12)
- 收藏
- 关注
原创 pytorch安装教程(未尝试,纯理论,后期测试)
采用Conda方式安装时,会自动安装pytorch和cuda工具包(本文小写的cuda指安装pytorch语句中带的cuda,并非英伟达官网下载的完整的CUDA,pytorch+cuda即可完成大多数训练。torchversion中有时会需要cuDNN,cuDNN应该需要完整的CUDA,需要时下载即可)。采用Wheel方式安装时,v2.0.0~v2.3.1中,从pytorch官网语句上得到的“表面结论”是不带cuda了,可能需要自行下载CUDA,(cuDNN的需求同上)。
2024-08-22 17:30:58
333
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人