2080Ti上SATS环境安装

首先要确保nvcc版本和要装的cuda版本一致,从而能够成功安装apex库,查看当前nvcc的版本:

nvcc --version

一开始我的版本是10.0,去官网上下载了10.2的更新
在这里插入图片描述
然后就可以开始配置环境了,首先创建虚拟环境

conda create -n SATS python=3.8 -y
conda activate SATS

然后安装pytorch,版本好像不能太高,不然apex库会安装失败,一开始我装的1.10.0版本pytorch就不行

conda install pytorch==1.8.1 torchvision cudatoolkit=10.2 -c pytorch

然后安装apex库

cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

中间可能会报一些错,最后只要显示如下信息就说明安装成功了
在这里插入图片描述
接着只要安装一些剩余的依赖环境即可

pip install tqdm
pip install matplotlib
pip install einops
pip install timm
pip install imgviz
pip install opencv-python-headless
pip install tensorboardX
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值