NVIDIA GTX 3090安装GPU版pytorch,mxnet和autogluon Ubuntu20.04
说明:Ubuntu20.04,NVIDIA GTX 3090的cudatoolkit必须大于11.0。另外,截止2021年11月02日,pytorch和mxnet均支持的cudatoolkit版本也是11.0,对于的pytorch版本只有1.7.1,mxnet版本mxnet_cu110
安装
新建并且激活环境
conda create -n autogluon python=3.8
conda activate autogluon
安装pytorch1.7.1
conda install p
原创
2021-11-02 14:21:53 ·
1156 阅读 ·
0 评论