查看CUDA的版本:
nvcc -V
查看所有已安装的CUDA版本:
ls /usr/local/ | grep cuda
切换使用的CUDA版本:
vim ~/.bashrc
改最后的这里:
改成想使用的就行 ,最后:
source ~/.bashrc
安包指定镜像源
阿里源
pip install 包名 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
清华源
pip install package_name -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn