解决Jetson Xavier NX上Invalid CUDA '--device 0' requested等问题
问题1:AssertionError: Invalid CUDA ‘–device 0’ requested, use ‘–device cpu’ or pass valid CUDA device(s)
主要原因:显卡驱动版本、CUDA版本、Pytorch版本不对应导致
以及安装了torch和torchvision,但是无法调用GPU,进一步缩小范围,在终端输入:
// $ python
import torch
torch.__version__
//显示torch版本号
print(torch.cuda.is_available());
//显示为Flase
说明安装的torch为CPU版本,卸载重新安装对应的GPU版本。
在下方网址查看符合CUDA和驱动的pytorch
https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048
若不知道jetpack版本,可以通过jtop查看,我的版本是:jetpack 4.6 [L4T 32.6.1]
所以下载“torch-1.8.0-cp36-cp36m-linux_aarch64.whl”以及“torchvision-0.9.0a0+01dfa8e-cp36-cp36m-linux_aarch64.whl”
运行下方命令进行安装:
pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl
pip install torchvision-0.9.0a0+01dfa8e-cp36-cp36m-linux_aarch64.whl
由于官网下载页面有时候打不开,可以通过百度网盘链接下载:
链接: https://pan.baidu.com/s/1T9fe8TYWC84cil3jU7XlrA?pwd=nyhx 提取码: nyhx
问题2: “Illegal instruction(cpre dumped)”错误
sudo gedit ~/.bashrc
//将“export OPENBLAS_CORETYPE=ARMV8”加入到最后
source ~/.bashrc
//更新bashrc文件
记录
python http局域网文件传输共享方法
python3:
python -m http.server 12324
//开启端口
http://192.xxx.x.xxx:12324
//接收端口
python2
python -m SimpleHTTPServer 12324
//开启端口
http://192.xxx.x.xxx:12324
//接收端口
使用清华源下载
pip install xxx -i https://pypi.tuna.tsinghua.edu.cn/simple