pytracking安装及运行3090Ubuntu18+cuda11(1)

`由于3090只能安装cuda11以上的版本,与作者描述的步骤有些不同,所以总结一下问题,避免看到的小伙伴走弯路

安装步骤:

`提示:与作者提供的顺序大致相同,但需要更改cuda版本

conda create --name pytracking python=3.7
conda activate pytracking
2.
conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
*如果添加清华镜像,去掉-c python速度更快,且不易失败。(-c python指定下载路径为pytorch官网)
另外,由于torch.rfft不能在torch1.7以上的版本运行,下面会介绍这个错误。
3.
conda install matplotlib pandas tqdm
pip install opencv-python visdom tb-nightly scikit-image tikzplotlib gdown
4.
conda install cython
pip install pycocotools
pip install lvis
5.
sudo apt-get install ninja-build
6.
`pip install spatial-correlation-sampler
7.
sudo apt-get install libturbojpeg
pip install jpeg4py

遇到的问题及解决方法

问题1./bin/sh: 1: /usr/local/cuda:/bin/nvcc: not found

解决:在终端运行:export CUDA_HOME=/usr/local/cuda-11.0,然后再次运行代码。

问题2.nvcc fatal : Unsupported gpu architecture ‘compute_86’

解决:在终端运行:export TORCH_CUDA_ARCH_LIST=“8.0”
*提示:有的解决方法为export TORCH_CUDA_ARCH_LIST=“7.5”,这个是不可以解决问题的,而且会引来新的错误:CUDA Error: no kernel image is available for execution on device。

问题3.module ‘torch’ has no attribute ‘rfft’

解决:安装torch1.7.1,上面提到了。torch1.8及以上都没有rfft。


  • 3
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值