解决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