RTX3090 + cuda 11.1 + torch1.9.0 安装 MinkowskiEngine
创建conda环境
conda create -n mink python=3.8
source activate mink
安装torch1.9.0 + torchvision
可到此处下载whl文件(也可以直接pip安装)
pip install torch-1.9.0+cu111-cp38-cp38-linux_x86_64.whl
然后安装torchvision
pip install torchvision
安装完后要看torch对应的cuda版本是否为cuda11.1
import tor
原创
2021-08-22 10:22:12 ·
1455 阅读 ·
2 评论